diff --git a/main.py b/main.py
index d172a49..50b847d 100755
--- a/main.py
+++ b/main.py
@@ -22,7 +22,7 @@ if __name__ == "__main__":
     positions = parser.get_positions()
 
     if args.visualize:
-        screen_dimensions = (1024*1.5, 1024*1.5 * HEIGHT_MM/WIDTH_MM)
+        screen_dimensions = (1024*1.5, 1024*1.5 * args.height/args.width)
         visualizer = Visualizer(positions, screen_dimensions, (args.width, args.height))
         visualizer.visualize()