diff --git a/mm_agent_example.py b/mm_agent_example.py
index 2dfb6e96..3177939c 100644
--- a/mm_agent_example.py
+++ b/mm_agent_example.py
@@ -1,7 +1,7 @@
 from swarms.agents import MultiModalAgent
 
 load_dict = {
-    "ImageCaptioning": "check"
+    "ImageCaptioning": "cuda"
 }
 
 node = MultiModalAgent(load_dict)