diff --git a/docs/extras/modules/agents/tools/how_to/custom_tools.ipynb b/docs/extras/modules/agents/tools/how_to/custom_tools.ipynb
index e4efc721d4..ecd1245046 100644
--- a/docs/extras/modules/agents/tools/how_to/custom_tools.ipynb
+++ b/docs/extras/modules/agents/tools/how_to/custom_tools.ipynb
@@ -934,7 +934,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "from langchain.schema import ToolException\n",
+    "from langchain.tools.base import ToolException\n",
     "\n",
     "from langchain import SerpAPIWrapper\n",
     "from langchain.agents import AgentType, initialize_agent\n",