diff --git a/docs/docs/use_cases/extraction.ipynb b/docs/docs/use_cases/extraction.ipynb
index e36586edacc..167d8c47022 100644
--- a/docs/docs/use_cases/extraction.ipynb
+++ b/docs/docs/use_cases/extraction.ipynb
@@ -430,7 +430,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 10,
+   "execution_count": null,
    "id": "64650362",
    "metadata": {},
    "outputs": [
@@ -452,8 +452,8 @@
     "from langchain.prompts import (\n",
     "    PromptTemplate,\n",
     ")\n",
+    "from langchain_core.pydantic_v1 import BaseModel, Field, validator\n",
     "from langchain_openai import OpenAI\n",
-    "from pydantic import BaseModel, Field, validator\n",
     "\n",
     "\n",
     "class Person(BaseModel):\n",
@@ -531,8 +531,8 @@
     "from langchain.prompts import (\n",
     "    PromptTemplate,\n",
     ")\n",
+    "from langchain_core.pydantic_v1 import BaseModel, Field, validator\n",
     "from langchain_openai import OpenAI\n",
-    "from pydantic import BaseModel, Field, validator\n",
     "\n",
     "\n",
     "# Define your desired data structure.\n",