diff --git a/agent.py b/agent.py
index ca478f0..f4dfb43 100644
--- a/agent.py
+++ b/agent.py
@@ -1,7 +1,7 @@
 import json
 
 from llm.base import LLMBase
-from mcp.client import MCPClient
+from client import MCPClient
 from config import MCP_URL