diff --git a/docs/docs/modules/agents/how_to/agent_iter.ipynb b/docs/docs/modules/agents/how_to/agent_iter.ipynb
index 618ed17e1e..d1f7fe0fce 100644
--- a/docs/docs/modules/agents/how_to/agent_iter.ipynb
+++ b/docs/docs/modules/agents/how_to/agent_iter.ipynb
@@ -22,7 +22,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "import pydantic\n",
+    "from pydantic.v1 import BaseModel, Field\n",
     "from langchain.agents import AgentType, initialize_agent\n",
     "from langchain.agents.tools import Tool\n",
     "from langchain.chains import LLMMathChain\n",
@@ -65,12 +65,12 @@
     "primes = {998: 7901, 999: 7907, 1000: 7919}\n",
     "\n",
     "\n",
-    "class CalculatorInput(pydantic.BaseModel):\n",
-    "    question: str = pydantic.Field()\n",
+    "class CalculatorInput(BaseModel):\n",
+    "    question: str = Field()\n",
     "\n",
     "\n",
-    "class PrimeInput(pydantic.BaseModel):\n",
-    "    n: int = pydantic.Field()\n",
+    "class PrimeInput(BaseModel):\n",
+    "    n: int = Field()\n",
     "\n",
     "\n",
     "def is_prime(n: int) -> bool:\n",