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",