diff --git a/swarms/agents/utils/manager.py b/swarms/agents/utils/manager.py
index 79aacf05..8eb383a7 100644
--- a/swarms/agents/utils/manager.py
+++ b/swarms/agents/utils/manager.py
@@ -3,7 +3,7 @@ from typing import Dict, Optional
 
 from langchain.agents.agent import AgentExecutor
 from langchain.callbacks.manager import CallbackManager
-from langchain.callbacks import set_handler
+from langchain.callbacks.base import set_handler
 from langchain.chains.conversation.memory import ConversationBufferMemory
 from langchain.memory.chat_memory import BaseChatMemory