diff --git a/scripts/6_train_Quick96.sh b/scripts/6_train_Quick96.sh
index c958f1b..507ef89 100755
--- a/scripts/6_train_Quick96.sh
+++ b/scripts/6_train_Quick96.sh
@@ -4,8 +4,8 @@ source env.sh
 $DFL_PYTHON "$DFL_SRC/main.py" train \
     --training-data-src-dir "$DFL_WORKSPACE/data_src/aligned" \
     --training-data-dst-dir "$DFL_WORKSPACE/data_dst/aligned" \
-    --pretraining-data-dir "$DFL_ROOT/pretrain_CelebA" \
-    --pretrained-model-dir "$DFL_ROOT/pretrain_Quick96" \
+    --pretraining-data-dir "$DFL_SRC/pretrain_CelebA" \
+    --pretrained-model-dir "$DFL_SRC/pretrain_Quick96" \
     --model-dir "$DFL_WORKSPACE/model" \
     --model Quick96
 
diff --git a/scripts/6_train_SAEHD.sh b/scripts/6_train_SAEHD.sh
index 10ea707..b82aceb 100755
--- a/scripts/6_train_SAEHD.sh
+++ b/scripts/6_train_SAEHD.sh
@@ -4,7 +4,7 @@ source env.sh
 $DFL_PYTHON "$DFL_SRC/main.py" train \
     --training-data-src-dir "$DFL_WORKSPACE/data_src/aligned" \
     --training-data-dst-dir "$DFL_WORKSPACE/data_dst/aligned" \
-    --pretraining-data-dir "$DFL_ROOT/pretrain_CelebA" \
+    --pretraining-data-dir "$DFL_SRC/pretrain_CelebA" \
     --model-dir "$DFL_WORKSPACE/model" \
     --model SAEHD