diff --git a/tests/test_router.py b/tests/test_router.py
index 7142ed1..cf0089a 100644
--- a/tests/test_router.py
+++ b/tests/test_router.py
@@ -4,7 +4,7 @@ import pytest
 from unittest.mock import MagicMock, patch
 
 from src.daemon.router import (
-    AgentRouter, AgentProfile, LLMDriver, RouteDecision, KNOWN_CAPABILITIES,
+    AgentRouter, AgentProfile, RouteDecision, KNOWN_CAPABILITIES,
 )