diff --git a/rl_coach/coach.py b/rl_coach/coach.py
index 3c5c1bd..00e8b97 100644
--- a/rl_coach/coach.py
+++ b/rl_coach/coach.py
@@ -663,6 +663,10 @@ class CoachLauncher(object):
             evaluation_worker.terminate()
 
 
-if __name__ == "__main__":
+def main():
     launcher = CoachLauncher()
     launcher.launch()
+
+
+if __name__ == "__main__":
+    main()