diff --git a/AIDungeon_2.ipynb b/AIDungeon_2.ipynb
index 231d2bc..a71ed8a 100644
--- a/AIDungeon_2.ipynb
+++ b/AIDungeon_2.ipynb
@@ -45,7 +45,7 @@
         "colab": {}
       },
       "source": [
-        "!git clone --depth 1 https://github.com/nickwalton/AIDungeon/\n",
+        "!git clone --depth 1 --branch master https://github.com/nickwalton/AIDungeon/\n",
         "%cd AIDungeon\n",
         "!./install.sh\n",
         "from IPython.display import clear_output \n",