diff --git a/docs/docs/how_to/tool_calling.ipynb b/docs/docs/how_to/tool_calling.ipynb
index 70faf4ce42..214d99557e 100644
--- a/docs/docs/how_to/tool_calling.ipynb
+++ b/docs/docs/how_to/tool_calling.ipynb
@@ -179,7 +179,7 @@
     "    b: Annotated[int, ..., \"Second integer\"]\n",
     "\n",
     "\n",
-    "class multiply(BaseModel):\n",
+    "class multiply(TypedDict):\n",
     "    \"\"\"Multiply two integers.\"\"\"\n",
     "\n",
     "    a: Annotated[int, ..., \"First integer\"]\n",