diff --git a/train.py b/train.py
index 8abb8cd..96cbc8d 100644
--- a/train.py
+++ b/train.py
@@ -118,5 +118,5 @@ def main2():
         break
 
 if __name__ == "__main__":
-    main2()
+    main()