diff --git a/src/grovers_visualizer/cli.py b/src/grovers_visualizer/cli.py
index 202623d..30497e9 100644
--- a/src/grovers_visualizer/cli.py
+++ b/src/grovers_visualizer/cli.py
@@ -73,7 +73,3 @@ def run_cli(args: Args) -> None:
 
     fig.canvas.mpl_disconnect(cid)
     plt.ioff()
-
-
-if __name__ == "__main__":
-    main()
diff --git a/src/grovers_visualizer/main.py b/src/grovers_visualizer/main.py
index b322bd6..409f606 100644
--- a/src/grovers_visualizer/main.py
+++ b/src/grovers_visualizer/main.py
@@ -17,3 +17,7 @@ def main() -> None:
         run_dpg_ui(args)
     else:
         run_cli(args)
+
+
+if __name__ == "__main__":
+    main()