diff --git a/tests/unit/test_classify_outcome.py b/tests/unit/test_classify_outcome.py
index 0cfdb97..c26ca1a 100644
--- a/tests/unit/test_classify_outcome.py
+++ b/tests/unit/test_classify_outcome.py
@@ -3,7 +3,7 @@ from typing import Optional
 
 pytestmark = pytest.mark.unit
 
-from src.daemon.spawner import Spawner
+from src.daemon.spawner import AgentSpawner as Spawner
 
 
 class TestClassifyCompleted: