diff --git a/docs/modules/agents/tools/custom_tools.ipynb b/docs/modules/agents/tools/custom_tools.ipynb
index 841cee6f1aafbd..2e2fa6f7f82f39 100644
--- a/docs/modules/agents/tools/custom_tools.ipynb
+++ b/docs/modules/agents/tools/custom_tools.ipynb
@@ -97,7 +97,7 @@
     "from pydantic import BaseModel, Field\n",
     "\n",
     "class CalculatorInput(BaseModel):\n",
-    "    query: str = Field(description=\"should be a math expression\")\n",
+    "    question: str = Field()\n",
     "        \n",
     "\n",
     "tools.append(\n",