diff --git a/scripts/5_XSeg_train.sh b/scripts/5_XSeg_train.sh
index 9965724..e2ae0b5 100755
--- a/scripts/5_XSeg_train.sh
+++ b/scripts/5_XSeg_train.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" \