diff --git a/docs/docs/use_cases/extraction.ipynb b/docs/docs/use_cases/extraction.ipynb
index 356445c676..fa3695cab2 100644
--- a/docs/docs/use_cases/extraction.ipynb
+++ b/docs/docs/use_cases/extraction.ipynb
@@ -379,8 +379,8 @@
     }
    ],
    "source": [
-    "from typing import Optional, List\n",
-    "from pydantic import BaseModel, Field\n",
+    "from typing import Optional\n",
+    "from langchain.pydantic_v1 import BaseModel\n",
     "from langchain.chains import create_extraction_chain_pydantic\n",
     "\n",
     "# Pydantic data class\n",