diff --git a/tests/test_agent_message_routing.py b/tests/test_agent_message_routing.py
index 6d0575918..3e83d88c5 100644
--- a/tests/test_agent_message_routing.py
+++ b/tests/test_agent_message_routing.py
@@ -3,7 +3,8 @@ from concurrent.futures import Future
 from dataclasses import replace
 from unittest.mock import AsyncMock, Mock, patch
 
-from app.agent.orchestrator import AgentManagerQueueFullError, MoviePilotAgent
+from app.agent.orchestrator import MoviePilotAgent
+from app.agent.session import AgentManagerQueueFullError
 from app.agent.tools.impl.ask_user_choice import (
     AskUserChoiceTool,
     UserChoiceOptionInput,