diff --git a/mm_agent_example.py b/mm_agent_example.py
index 0f8758ae..3177939c 100644
--- a/mm_agent_example.py
+++ b/mm_agent_example.py
@@ -1,7 +1,7 @@
 from swarms.agents import MultiModalAgent
 
 load_dict = {
-    "ImageCaptioning": "cudo"
+    "ImageCaptioning": "cuda"
 }
 
 node = MultiModalAgent(load_dict)
diff --git a/omnimodal_agent.py b/omnimodal_agent.py
index 07c0a76b..03fc81dd 100644
--- a/omnimodal_agent.py
+++ b/omnimodal_agent.py
@@ -1,4 +1,7 @@
-from swarms import OmniModalAgent, OpenAIChat
+from langchain.
+from swarms.agents import OmniModalAgent
+
+
 
 llm = OpenAIChat()
 
diff --git a/swarms/agents/__init__.py b/swarms/agents/__init__.py
index 11832e55..467e67f2 100644
--- a/swarms/agents/__init__.py
+++ b/swarms/agents/__init__.py
@@ -3,7 +3,7 @@
 #agents
 # from swarms.agents.profitpilot import ProfitPilot
 from swarms.agents.aot import AoTAgent
-from swarms.agents.multi_modal_visual_agent import MultiModalAgent
+# from swarms.agents.multi_modal_visual_agent import MultiModalAgent
 from swarms.agents.omni_modal_agent import OmniModalAgent