diff --git a/agent/agent_config.py b/agent/agent_config.py
index 41b0436..bd12fc5 100644
--- a/agent/agent_config.py
+++ b/agent/agent_config.py
@@ -233,7 +233,7 @@ class AgentConfig:
 
         from utils.settings import LANGFUSE_ENABLED
         if LANGFUSE_ENABLED:
-            from langfuse.callback import CallbackHandler as LangfuseCallbackHandler
+            from langfuse.langchain import CallbackHandler as LangfuseCallbackHandler
             langfuse_handler = LangfuseCallbackHandler(
                 trace_id=self.trace_id,
                 session_id=self.session_id,