diff --git a/swarms/agents/omni_modal_agent.py b/swarms/agents/omni_modal_agent.py
index 4f9e4637..e5715cbf 100644
--- a/swarms/agents/omni_modal_agent.py
+++ b/swarms/agents/omni_modal_agent.py
@@ -1,4 +1,4 @@
-from langchain_community.base_language import BaseLanguageModel
+from langchain_core.base_language import BaseLanguageModel
 from langchain_experimental.autonomous_agents.hugginggpt.repsonse_generator import (
     load_response_generator,
 )