diff --git a/gain-viz/app.py b/gain-viz/app.py
index dd7182a..9ec276f 100644
--- a/gain-viz/app.py
+++ b/gain-viz/app.py
@@ -215,7 +215,7 @@ def get_gains():
     })
 
 # ----------------- Main -----------------
-if __name__ == '__main__':
+def main():
     # Ensure placeholder image exists
     if not os.path.exists(PLOT_PATH):
         plt.figure()