diff --git a/gpt_env/exec_python_code.py b/gpt_env/exec_python_code.py
index 69cf7fd..c4fb785 100644
--- a/gpt_env/exec_python_code.py
+++ b/gpt_env/exec_python_code.py
@@ -54,4 +54,5 @@ def test2():
     result = run_code(code_str1, create_namespace())
     print(result)
 
-test2()
\ No newline at end of file
+if __name__ == "__main__":
+    test2()