diff --git a/src/agent.py b/src/agent.py
index c0b50ba..46f5c3b 100644
--- a/src/agent.py
+++ b/src/agent.py
@@ -1,7 +1,7 @@
 import argparse
-from langchain.agents import create_openai_functions_agent, AgentExecutor
+from langchain.agents.openai_functions import create_openai_functions_agent
+from langchain.agents import AgentExecutor
 from langchain_core.prompts import ChatPromptTemplate
-from langchain_core.messages import HumanMessage
 from langchain_ollama import Ollama
 from src.utils import load_faq_to_chroma, search_course_docs, fetch_course_meta