diff --git a/notebooks/unit2/unit2.ipynb b/notebooks/unit2/unit2.ipynb
index ea808f6..7f3c3da 100644
--- a/notebooks/unit2/unit2.ipynb
+++ b/notebooks/unit2/unit2.ipynb
@@ -280,6 +280,7 @@
         "import random\n",
         "import imageio\n",
         "import os\n",
+        "import tqdm\n",
         "\n",
         "import pickle5 as pickle\n",
         "from tqdm.notebook import tqdm"