diff --git a/docs/docs/modules/model_io/output_parsers/types/retry.ipynb b/docs/docs/modules/model_io/output_parsers/types/retry.ipynb
index 8703a19b644a7..6a1a7f94a92aa 100644
--- a/docs/docs/modules/model_io/output_parsers/types/retry.ipynb
+++ b/docs/docs/modules/model_io/output_parsers/types/retry.ipynb
@@ -24,8 +24,8 @@
     "from langchain.prompts import (\n",
     "    PromptTemplate,\n",
     ")\n",
-    "from langchain_openai import ChatOpenAI, OpenAI\n",
-    "from pydantic import BaseModel, Field"
+    "from langchain_core.pydantic_v1 import BaseModel, Field\n",
+    "from langchain_openai import ChatOpenAI, OpenAI"
    ]
   },
   {