diff --git a/labs/ai-foundry-model-gateway/foundry-ai-gateway.ipynb b/labs/ai-foundry-model-gateway/foundry-ai-gateway.ipynb
index d2706374..45e3eb7c 100644
--- a/labs/ai-foundry-model-gateway/foundry-ai-gateway.ipynb
+++ b/labs/ai-foundry-model-gateway/foundry-ai-gateway.ipynb
@@ -385,7 +385,7 @@
     "response = openai_client.responses.create(\n",
     "    input=[{\"role\": \"user\", \"content\": \"Tell me what you can help with. and tell me a funny short joke.\"}],\n",
     "    conversation=conversation_id,\n",
-    "    extra_body={\"agent\": {\"name\": agent.name, \"type\": \"agent_reference\"}},\n",
+    "    extra_body={\"agent_reference\": {\"name\": agent.name, \"type\": \"agent_reference\"}},\n",
     ")\n",
     "\n",
     "utils.print_info(f\"📝 Response: {response.output_text}\")"