diff --git a/docs/extras/use_cases/extraction.ipynb b/docs/extras/use_cases/extraction.ipynb
index 628026127a4..b607029751d 100644
--- a/docs/extras/use_cases/extraction.ipynb
+++ b/docs/extras/use_cases/extraction.ipynb
@@ -443,7 +443,7 @@
     }
    ],
    "source": [
-    "from typing import Sequence\n",
+    "from typing import Sequence, Optional\n",
     "from langchain.prompts import (\n",
     "    PromptTemplate,\n",
     "    ChatPromptTemplate,\n",