diff --git a/colabs/image_classification.ipynb b/colabs/image_classification.ipynb
index d03d302146..b0b58f5025 100644
--- a/colabs/image_classification.ipynb
+++ b/colabs/image_classification.ipynb
@@ -6,7 +6,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "%% writefile config.yml\n",
+    "%%writefile config.yml\n",
     "task: image_classification # do not change\n",
     "base_model: google/vit-base-patch16-224 # the model to be used from hugging face hub\n",
     "project_name: autotrain-image-classification-model # the name of the project, must be unique\n",
@@ -47,6 +47,7 @@
     "HF_USERNAME = userdata.get('HF_USERNAME')\n",
     "HF_TOKEN = userdata.get('HF_TOKEN')\n",
     "os.environ['HF_USERNAME'] = HF_USERNAME\n",
+    "\n",
     "os.environ['HF_TOKEN'] = HF_TOKEN\n",
     "!autotrain --config config.yml"
    ]