diff --git a/docs/source/notebooks/tool_usage/benchmark_all_tasks.ipynb b/docs/source/notebooks/tool_usage/benchmark_all_tasks.ipynb
index 9b64746..5bfabd9 100644
--- a/docs/source/notebooks/tool_usage/benchmark_all_tasks.ipynb
+++ b/docs/source/notebooks/tool_usage/benchmark_all_tasks.ipynb
@@ -152,7 +152,7 @@
     "    [\n",
     "        (\n",
     "            \"human\",\n",
-    "            \"{instructions}\\n{input}\",\n",
+    "            \"{instructions}\\n{question}\",\n",
     "        ),  # Populated from task.instructions automatically\n",
     "        MessagesPlaceholder(\"agent_scratchpad\"),  # Workspace for the agent\n",
     "    ]\n",
@@ -161,7 +161,7 @@
     "with_system_message_prompt = ChatPromptTemplate.from_messages(\n",
     "    [\n",
     "        (\"system\", \"{instructions}\"),\n",
-    "        (\"human\", \"{input}\"),  # Populated from task.instructions automatically\n",
+    "        (\"human\", \"{question}\"),  # Populated from task.instructions automatically\n",
     "        MessagesPlaceholder(\"agent_scratchpad\"),  # Workspace for the agent\n",
     "    ]\n",
     ")"