diff --git a/playground/demos/developer_swarm/main.py b/playground/demos/developer_swarm/main.py
index 793936a2..54170985 100644
--- a/playground/demos/developer_swarm/main.py
+++ b/playground/demos/developer_swarm/main.py
@@ -27,8 +27,7 @@ api_key = os.getenv("OPENAI_API_KEY")
 
 
 TASK = """
-code
-
+CODE
 
 """
 
@@ -38,12 +37,12 @@ llm = OpenAIChat(openai_api_key=api_key, max_tokens=5000)
 
 # Documentation agent
 documentation_agent = Agent(
-    llm=llm, sop=DOCUMENTATION_SOP, max_loops=1, multi_modal=True
+    llm=llm, sop=DOCUMENTATION_SOP, max_loops=1,
 )
 
 
 # Tests agent
-tests_agent = Agent(llm=llm, sop=TEST_SOP, max_loops=2, multi_modal=True)
+tests_agent = Agent(llm=llm, sop=TEST_SOP, max_loops=2,)
 
 
 # Run the documentation agent