diff --git a/secgym/agents/baseline_agent.py b/secgym/agents/baseline_agent.py
index 0246fd0..cd26b1b 100644
--- a/secgym/agents/baseline_agent.py
+++ b/secgym/agents/baseline_agent.py
@@ -274,7 +274,7 @@ def reset(self, change_seed=True):
 
         self.step_count = 0
         sys_prompt = BASE_PROMPT
-        if "o1" in self.config_list[0]['model'] or "o3" in self.config_list[0]['model'] or "o4" in self.config_list[0]['model'] or "meta-llama" in self.config_list[0]['model'] or "qwen3" in self.config_list[0]['model']:
+        if "o1" in self.config_list[0]['model'] or "o3" in self.config_list[0]['model'] or "o4" in self.config_list[0]['model'] or "meta-llama" in self.config_list[0]['model']:
             sys_prompt = O1_PROMPT
         elif "r1" in self.config_list[0]['model'] or "R1" in self.config_list[0]['model'] or "qwen3" in self.config_list[0]['model']:
             sys_prompt = R1_PROMPT