diff --git a/map2map/main.py b/map2map/main.py
index da06840..a8fa1dc 100644
--- a/map2map/main.py
+++ b/map2map/main.py
@@ -11,3 +11,7 @@ def main():
         train.node_worker(args)
     elif args.mode == 'test':
         test.test(args)
+
+
+if __name__ == '__main__':
+    main()