diff --git a/tests/test_codex_execution_paths.py b/tests/test_codex_execution_paths.py
index 892053831b..b3550d883c 100644
--- a/tests/test_codex_execution_paths.py
+++ b/tests/test_codex_execution_paths.py
@@ -80,7 +80,7 @@ class _Codex401ThenSuccessAgent(run_agent.AIAgent):
         type(self).refresh_attempts += 1
         return True
 
-    def run_conversation(self, user_message: str, conversation_history=None):
+    def run_conversation(self, user_message: str, conversation_history=None, task_id=None):
         calls = {"api": 0}
 
         def _fake_api_call(api_kwargs):
@@ -90,7 +90,7 @@ class _Codex401ThenSuccessAgent(run_agent.AIAgent):
             return _codex_message_response("Recovered via refresh")
 
         self._interruptible_api_call = _fake_api_call
-        return super().run_conversation(user_message, conversation_history=conversation_history)
+        return super().run_conversation(user_message, conversation_history=conversation_history, task_id=task_id)
 
 
 def test_cron_run_job_codex_path_handles_internal_401_refresh(monkeypatch):