diff --git a/scripts/5_XSeg_data_src_trained_mask_remove.sh b/scripts/5_XSeg_data_src_trained_mask_remove.sh
index 14b9a98..36f1994 100755
--- a/scripts/5_XSeg_data_src_trained_mask_remove.sh
+++ b/scripts/5_XSeg_data_src_trained_mask_remove.sh
@@ -2,5 +2,5 @@
 cd "$(dirname $0)/.."
 source scripts/env.sh
 
-python "$DFL_MAIN" xseg remove \
+$DFL_PYTHON "$DFL_MAIN" xseg remove \
   --input-dir "$WORKSPACE/data_src/aligned"