diff --git a/docs/docs/how_to/structured_output.ipynb b/docs/docs/how_to/structured_output.ipynb
index c00755650c1..66d961491d2 100644
--- a/docs/docs/how_to/structured_output.ipynb
+++ b/docs/docs/how_to/structured_output.ipynb
@@ -165,6 +165,8 @@
     }
    ],
    "source": [
+    "from typing import Optional\n",
+    "\n",
     "from typing_extensions import Annotated, TypedDict\n",
     "\n",
     "\n",
@@ -206,10 +208,10 @@
     {
      "data": {
       "text/plain": [
-        "{'setup': 'Why was the cat sitting on the computer?',\n",
-        " 'punchline': 'Because it wanted to keep an eye on the mouse!',\n",
-        " 'rating': 7}"
-       ]
+       "{'setup': 'Why was the cat sitting on the computer?',\n",
+       " 'punchline': 'Because it wanted to keep an eye on the mouse!',\n",
+       " 'rating': 7}"
+      ]
      },
      "execution_count": 4,
      "metadata": {},