diff --git a/src/openjarvis/agents/executor.py b/src/openjarvis/agents/executor.py
index 1b8eb8c0..6a5fe6e0 100644
--- a/src/openjarvis/agents/executor.py
+++ b/src/openjarvis/agents/executor.py
@@ -261,7 +261,7 @@ class AgentExecutor:
         if router_policy_key and self._system:
             try:
                 from openjarvis.core.registry import RouterPolicyRegistry
-                from openjarvis.learning.routing.types import (
+                from openjarvis.learning.routing.router import (
                     build_routing_context,
                 )