diff --git a/scripts/make_colab_zip.py b/scripts/make_colab_zip.py
index 755d708..c387c2e 100644
--- a/scripts/make_colab_zip.py
+++ b/scripts/make_colab_zip.py
@@ -4,8 +4,8 @@ pretrain mode (default):
   - src/                               (all Python modules)
   - scripts/pretrain.py
   - requirements.txt
-  - data/processed/mcgill/train/*.pt   (remapped from data/processed/train/)
-  - data/processed/mcgill/val/*.pt     (remapped from data/processed/val/)
+  - data/processed/mcgill/train/*.pt
+  - data/processed/mcgill/val/*.pt
 
 finetune mode:
   - src/                               (all Python modules)
@@ -57,8 +57,8 @@ MODE_SCRIPTS: dict[str, list[str]] = {
 
 # Local dir → arc path inside zip
 PRETRAIN_DATA: list[tuple[Path, str]] = [
-    (ROOT / "data" / "processed" / "train", "data/processed/mcgill/train"),
-    (ROOT / "data" / "processed" / "val",   "data/processed/mcgill/val"),
+    (ROOT / "data" / "processed" / "mcgill" / "train", "data/processed/mcgill/train"),
+    (ROOT / "data" / "processed" / "mcgill" / "val",   "data/processed/mcgill/val"),
 ]
 
 FINETUNE_DATA: list[tuple[Path, str]] = [