diff --git a/nlsam/tests/test_scripts.sh b/nlsam/tests/test_scripts.sh
index 24a561cd..5df24ec5 100644
--- a/nlsam/tests/test_scripts.sh
+++ b/nlsam/tests/test_scripts.sh
@@ -13,6 +13,7 @@ function check_return_code()
 
 # Crop example dataset
 python -c 'import nibabel as nib; import numpy as np; d = nib.load("dwi.nii.gz").get_data(); nib.save(nib.Nifti1Image(d[70:80],np.eye(4)), "dwi.nii.gz")'
+python -c 'import nibabel as nib; import numpy as np; d = nib.load("mask.nii.gz").get_data(); nib.save(nib.Nifti1Image(d[70:80],np.eye(4)), "mask.nii.gz")'
 
 # Test on example dataset
 stabilizer dwi.nii.gz dwi_stab_localstd.nii.gz 1 sigma_localstd.nii.gz -m mask.nii.gz --bvals bvals --bvecs bvecs --noise_est local_std --smooth no_smoothing