diff --git a/colabs/sampling_tutorial.ipynb b/colabs/sampling_tutorial.ipynb
index 9563adf..3eb0f46 100644
--- a/colabs/sampling_tutorial.ipynb
+++ b/colabs/sampling_tutorial.ipynb
@@ -107,7 +107,7 @@
     "VARIANT = '2b-it' # @param ['2b', '2b-it', '7b', '7b-it'] {type:\"string\"}\n",
     "weights_dir = kagglehub.model_download(f'google/gemma/Flax/{VARIANT}')\n",
     "\n",
-    "ckpt_path = os.path.join(weights_dir, variant)\n",
+    "ckpt_path = os.path.join(weights_dir, VARIANT)\n",
     "vocab_path = os.path.join(weights_dir, 'tokenizer.model')"
    ]
   },