diff --git a/flash_notebooks/image_classification.ipynb b/flash_notebooks/image_classification.ipynb
index 342bf72c74..4c5766ea7d 100644
--- a/flash_notebooks/image_classification.ipynb
+++ b/flash_notebooks/image_classification.ipynb
@@ -267,7 +267,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "model = ImageClassifier.load_from_checkpoint(\"https://flash-weights.s3.amazonaws.com/image_classification_model.pt\")"
+    "model = ImageClassifier.load_from_checkpoint(\"https://flash-weights.s3.amazonaws.com/0.6.0/image_classification_model.pt\")"
    ]
   },
   {