diff --git a/main.py b/main.py
index ce01fca..c55537a 100644
--- a/main.py
+++ b/main.py
@@ -36,7 +36,7 @@ def main():
         )
     )
 
-    ImageUtil.save_image(image, "image.png")
+    ImageUtil.save_image(image, args.output)
 
 
 if __name__ == '__main__':