diff --git a/rdagent/app/data_science/loop.py b/rdagent/app/data_science/loop.py
index d64c2e2f..c99c198a 100644
--- a/rdagent/app/data_science/loop.py
+++ b/rdagent/app/data_science/loop.py
@@ -14,7 +14,7 @@ def main(
     checkout: bool | str | Path = True,
     step_n: int | None = None,
     loop_n: int | None = None,
-    all_duration: str | None = None,
+    timeout: str | None = None,
     competition="bms-molecular-translation",
     replace_timer=True,
     exp_gen_cls: str | None = None,
@@ -67,7 +67,7 @@ def main(
     if exp_gen_cls is not None:
         kaggle_loop.exp_gen = import_class(exp_gen_cls)(kaggle_loop.exp_gen.scen)
 
-    asyncio.run(kaggle_loop.run(step_n=step_n, loop_n=loop_n, all_duration=all_duration))
+    asyncio.run(kaggle_loop.run(step_n=step_n, loop_n=loop_n, all_duration=timeout))
 
 
 if __name__ == "__main__":
diff --git a/rdagent/log/server/app.py b/rdagent/log/server/app.py
index 807cd352..cca4e9fd 100644
--- a/rdagent/log/server/app.py
+++ b/rdagent/log/server/app.py
@@ -145,7 +145,7 @@ def upload_file():
         if loop_n:
             cmds += ["--loop_n", loop_n]
     if all_duration:
-        cmds += ["--all_duration", f"{all_duration}h"]
+        cmds += ["--timeout", f"{all_duration}h"]
 
     app.logger.info(f"Started process for {log_trace_path} with parameters: {cmds}")
     with stdout_path.open("w") as log_file: