diff --git a/agency_swarm/agency/agency.py b/agency_swarm/agency/agency.py
index d1e45270..743739bf 100644
--- a/agency_swarm/agency/agency.py
+++ b/agency_swarm/agency/agency.py
@@ -636,7 +636,7 @@ def on_all_streams_end(cls):
                     cls.message_output = None
                     chatbot_queue.put("[end]")
 
-            def bot(original_message, history):
+            def bot(original_message, history, dropdown):
                 nonlocal attachments
                 nonlocal message_file_names
                 nonlocal recipient_agent
@@ -757,7 +757,7 @@ def bot(original_message, history):
             dropdown.change(handle_dropdown_change, dropdown)
             file_upload.change(handle_file_upload, file_upload)
             msg.submit(user, [msg, chatbot], [msg, chatbot], queue=False).then(
-                bot, [msg, chatbot], [msg, chatbot, dropdown]
+                bot, [msg, chatbot, dropdown], [msg, chatbot, dropdown]
             )
 
             # Enable queuing for streaming intermediate outputs
diff --git a/notebooks/os_models_with_astra_assistants_api.ipynb b/notebooks/os_models_with_astra_assistants_api.ipynb
index f3ff337f..cf7e4bb2 100644
--- a/notebooks/os_models_with_astra_assistants_api.ipynb
+++ b/notebooks/os_models_with_astra_assistants_api.ipynb
@@ -376,7 +376,7 @@
     "\n",
     "            return original_user_message, history + [[user_message, None]]\n",
     "\n",
-    "        def bot(original_message, history):\n",
+    "        def bot(original_message, history, dropdown):\n",
     "            nonlocal message_file_ids\n",
     "            nonlocal message_file_names\n",
     "            nonlocal recipient_agent\n",