diff --git a/src/crackling/utils/trainModel.py b/src/crackling/utils/trainModel.py
index 74c3ec7..d26f9c4 100644
--- a/src/crackling/utils/trainModel.py
+++ b/src/crackling/utils/trainModel.py
@@ -128,3 +128,6 @@ def main():
 		print('Using user specified arguments')
 		# Else parse args
 		trainModel(sys.argv[1:])
+
+if __name__ == '__main__':
+    main()