diff --git a/train.ipynb b/train.ipynb
index 3c9bedf..9dda892 100644
--- a/train.ipynb
+++ b/train.ipynb
@@ -112,7 +112,7 @@
       "outputs": [],
       "source": [
         "!pip install googledrivedownloader\n",
-        "!pip install timm\n",
+        "!pip install timm==0.5.4\n",
         "!wget -P SimSwap/arcface_model https://github.com/neuralchen/SimSwap/releases/download/1.0/arcface_checkpoint.tar"
       ]
     },
@@ -143,7 +143,7 @@
         "%mkdir /content/TrainingData\n",
         "!wget --load-cookies /tmp/cookies.txt \"https://docs.google.com/uc?export=download&confirm=$(wget --quiet --save-cookies /tmp/cookies.txt --keep-session-cookies --no-check-certificate 'https://docs.google.com/uc?export=download&id=19pWvdEHS-CEG6tW3PdxdtZ5QEymVjImc' -O- | sed -rn 's/.*confirm=([0-9A-Za-z_]+).*/\\1\\n/p')&id=19pWvdEHS-CEG6tW3PdxdtZ5QEymVjImc\" -O /content/TrainingData/vggface2_crop_arcfacealign_224.tar && rm -rf /tmp/cookies.txt\n",
         "%cd /content/\n",
-        "!tar -xzvf /content/TrainingData/vggface2_crop_arcfacealign_224.tar\n",
+        "!tar -xzvf /content/TrainingData/vggface2_crop_arcfacealign_224.tar -C /content/TrainingData\n ",
         "!rm /content/TrainingData/vggface2_crop_arcfacealign_224.tar"
       ]
     },