diff --git a/docs/source/examples/Output Widget.ipynb b/docs/source/examples/Output Widget.ipynb
index 70c9de2a2c..8a4d74cb9c 100644
--- a/docs/source/examples/Output Widget.ipynb	
+++ b/docs/source/examples/Output Widget.ipynb	
@@ -354,6 +354,7 @@
     "```python\n",
     "import threading\n",
     "import time\n",
+    "import itertools\n",
     "\n",
     "def run():\n",
     "    for i in itertools.count(0):\n",