diff --git a/expo/MCTS.py b/expo/MCTS.py
index 378474b4e..cfb21a61c 100644
--- a/expo/MCTS.py
+++ b/expo/MCTS.py
@@ -20,7 +20,11 @@ from metagpt.utils.common import read_json_file
 
 def initialize_di_root_node(state, reflection: bool = True):
     role = ResearchAssistant(
-        node_id="0", start_task_id=state["start_task_id"], use_reflection=reflection, role_dir=state["node_dir"]
+        node_id="0",
+        start_task_id=state["start_task_id"],
+        use_reflection=reflection,
+        role_dir=state["node_dir"],
+        role_timeout=state["role_timeout"],
     )
     return role, Node(parent=None, state=state, action=None, value=0)
 
diff --git a/expo/run_experiment.py b/expo/run_experiment.py
index bf90cb07a..71529b955 100644
--- a/expo/run_experiment.py
+++ b/expo/run_experiment.py
@@ -9,7 +9,7 @@ from expo.experimenter.experimenter import Experimenter
 from expo.experimenter.mcts import MCTSExperimenter
 
 
-def get_args():
+def get_args(cmd=True):
     parser = argparse.ArgumentParser()
     parser.add_argument("--name", type=str, default="")
     parser.add_argument(
@@ -22,7 +22,10 @@ def get_args():
     get_di_args(parser)
     get_mcts_args(parser)
     get_aug_exp_args(parser)
-    return parser.parse_args()
+    if cmd:
+        return parser.parse_args()
+    else:
+        return parser.parse_args("")
 
 
 def get_mcts_args(parser):