diff --git a/rl_unplugged/atari_dqn.ipynb b/rl_unplugged/atari_dqn.ipynb
index 0aa71e5..bd5a54e 100644
--- a/rl_unplugged/atari_dqn.ipynb
+++ b/rl_unplugged/atari_dqn.ipynb
@@ -126,7 +126,10 @@
         "gs_path = 'gs://rl_unplugged/atari'\n",
         "\n",
         "!mkdir -p {tmp_path}/{game}\n",
-        "!gsutil cp {gs_path}/{game}/run_{run}-00000-of-00001 {tmp_path}/{game}"
+        "\n",
+        "src = f'{gs_path}/{game}/run_{run}-00000-of-00100'\n",
+        "dest = f'{tmp_path}/{game}/run_{run}-00000-of-00001'\n",
+        "!gsutil cp {src} {dest}"
       ]
     },
     {
@@ -386,8 +389,8 @@
     "colab": {
       "collapsed_sections": [],
       "last_runtime": {
-        "build_target": "",
-        "kind": "local"
+        "build_target": "//learning/deepmind/dm_python:dm_notebook3",
+        "kind": "private"
       },
       "name": "RL Unplugged: Offline DQN - Atari",
       "provenance": [