diff --git a/docs/docs/how_to/custom_tools.ipynb b/docs/docs/how_to/custom_tools.ipynb
index 8becb03c7c9..4225569ab61 100644
--- a/docs/docs/how_to/custom_tools.ipynb
+++ b/docs/docs/how_to/custom_tools.ipynb
@@ -169,7 +169,7 @@
     "    return a * max(b)\n",
     "\n",
     "\n",
-    "multiply_by_max.args_schema.schema()"
+    "print(multiply_by_max.args_schema.model_json_schema())"
    ]
   },
   {
@@ -285,7 +285,7 @@
     "    return bar\n",
     "\n",
     "\n",
-    "foo.args_schema.schema()"
+    "print(foo.args_schema.model_json_schema())"
    ]
   },
   {