diff --git a/src/HowBadIsMyBatch.ipynb b/src/HowBadIsMyBatch.ipynb
index 1c75676e311..34aeb041bc5 100644
--- a/src/HowBadIsMyBatch.ipynb
+++ b/src/HowBadIsMyBatch.ipynb
@@ -234,6 +234,7 @@
    "outputs": [],
    "source": [
     "import zipfile\n",
+    "import os\n",
     "\n",
     "def unzip(zipFile, dstDir):\n",
     "    with zipfile.ZipFile(zipFile, 'r') as zip_ref:\n",