diff --git a/rdagent/components/coder/model_coder/model.py b/rdagent/components/coder/model_coder/model.py
index eca3b432..a8f700e9 100644
--- a/rdagent/components/coder/model_coder/model.py
+++ b/rdagent/components/coder/model_coder/model.py
@@ -115,6 +115,7 @@ PARAM_INIT_VALUE = {param_init_value}
                 dump_file_names=["execution_feedback_str.pkl", "execution_model_output.pkl"],
                 local_path=str(self.workspace_path),
                 env={},
+                code_dump_file_py_name="model_test",
             )
             if results is None:
                 raise RuntimeError(f"Error in running the model code: {log}")
diff --git a/rdagent/utils/env.py b/rdagent/utils/env.py
index 7c43d743..815dfcde 100644
--- a/rdagent/utils/env.py
+++ b/rdagent/utils/env.py
@@ -327,11 +327,12 @@ class DockerEnv(Env[DockerConf]):
         local_path: str | None = None,
         env: dict | None = None,
         running_extra_volume: dict | None = None,
+        code_dump_file_py_name: Optional[str] = None,
     ):
         """
         Dump the code into the local path and run the code.
         """
-        random_file_name = f"{uuid.uuid4()}.py"
+        random_file_name = f"{uuid.uuid4()}.py" if code_dump_file_py_name is None else f"{code_dump_file_py_name}.py"
         with open(os.path.join(local_path, random_file_name), "w") as f:
             f.write(code)
         entry = f"python {random_file_name}"