diff --git a/polygen/sample-pretrained.ipynb b/polygen/sample-pretrained.ipynb
index 353febd..e8c90b8 100644
--- a/polygen/sample-pretrained.ipynb
+++ b/polygen/sample-pretrained.ipynb
@@ -42,9 +42,7 @@
       },
       "outputs": [],
       "source": [
-        "!pip install tensorflow==1.15\n",
-        "!pip install dm-sonnet==1.36\n",
-        "!pip install tensor2tensor==1.14\n",
+        "!pip install tensorflow==1.15 dm-sonnet==1.36 tensor2tensor==1.14\n",
         "\n",
         "import time\n",
         "import numpy as np\n",
@@ -52,8 +50,11 @@
         "tf.logging.set_verbosity(tf.logging.ERROR)  # Hide TF deprecation messages\n",
         "import matplotlib.pyplot as plt\n",
         "\n",
-        "!git clone https://github.com/deepmind/deepmind-research.git deepmind_research\n",
-        "%cd deepmind_research/polygen\n",
+        "%cd /tmp\n",
+        "%rm -rf /tmp/deepmind_research\n",
+        "!git clone https://github.com/deepmind/deepmind-research.git \\\n",
+        "  /tmp/deepmind_research\n",
+        "%cd /tmp/deepmind_research/polygen\n",
         "import modules\n",
         "import data_utils"
       ]
diff --git a/polygen/training.ipynb b/polygen/training.ipynb
index b80e4ed..34d73af 100644
--- a/polygen/training.ipynb
+++ b/polygen/training.ipynb
@@ -42,9 +42,7 @@
       },
       "outputs": [],
       "source": [
-        "!pip install tensorflow==1.15\n",
-        "!pip install dm-sonnet==1.36\n",
-        "!pip install tensor2tensor==1.14\n",
+        "!pip install tensorflow==1.15 dm-sonnet==1.36 tensor2tensor==1.14\n",
         "\n",
         "import os\n",
         "import numpy as np\n",
@@ -52,8 +50,11 @@
         "tf.logging.set_verbosity(tf.logging.ERROR)  # Hide TF deprecation messages\n",
         "import matplotlib.pyplot as plt\n",
         "\n",
-        "!git clone https://github.com/deepmind/deepmind-research.git deepmind_research\n",
-        "%cd deepmind_research/polygen\n",
+        "%cd /tmp\n",
+        "%rm -rf /tmp/deepmind_research\n",
+        "!git clone https://github.com/deepmind/deepmind-research.git \\\n",
+        "  /tmp/deepmind_research\n",
+        "%cd /tmp/deepmind_research/polygen\n",
         "import modules\n",
         "import data_utils"
       ]