diff --git a/scripts/6_train_SAEHD.sh b/scripts/6_train_SAEHD.sh
index 2be58f0..e7adf2c 100755
--- a/scripts/6_train_SAEHD.sh
+++ b/scripts/6_train_SAEHD.sh
@@ -2,7 +2,7 @@
 cd "$(dirname $0)/.."
 source scripts/env.sh
 
-python "$DFL_MAIN" train \
+$DFL_PYTHON "$DFL_MAIN" train \
   --training-data-src-dir "$WORKSPACE/data_src/aligned" \
   --training-data-dst-dir "$WORKSPACE/data_dst/aligned" \
   --model-dir "$WORKSPACE/model" \