diff --git a/scripts/5_XSeg_data_src_mask_remove.sh b/scripts/5_XSeg_data_src_mask_remove.sh
index d467576..1d5e526 100755
--- a/scripts/5_XSeg_data_src_mask_remove.sh
+++ b/scripts/5_XSeg_data_src_mask_remove.sh
@@ -2,5 +2,5 @@
 cd "$(dirname $0)/.."
 source scripts/env.sh
 
-python "$DFL_MAIN" xseg remove_labels \
+$DFL_PYTHON "$DFL_MAIN" xseg remove_labels \
   --input-dir "$WORKSPACE/data_src/aligned"