diff --git a/examples/BraTS/data_prep/mlcube/mlcube.yaml b/examples/BraTS/data_prep/mlcube/mlcube.yaml
index 2cdfcabec..e05b3e184 100644
--- a/examples/BraTS/data_prep/mlcube/mlcube.yaml
+++ b/examples/BraTS/data_prep/mlcube/mlcube.yaml
@@ -8,7 +8,7 @@ platform:
 
 docker:
   # Image name.
-  image: mlcommons/fets_data-prep
+  image: mlcommons/fets_data-prep-v2
   # Docker build context relative to $MLCUBE_ROOT. Default is `build`.
   build_context: "../project"
   # Docker file name within docker build context, default is `Dockerfile`.
diff --git a/examples/BraTS/data_prep/project/sanity_check.py b/examples/BraTS/data_prep/project/sanity_check.py
index 68c4cfe0b..5eb0aa6a7 100644
--- a/examples/BraTS/data_prep/project/sanity_check.py
+++ b/examples/BraTS/data_prep/project/sanity_check.py
@@ -17,6 +17,7 @@ def check_subject_validity(subject_dir):
     subject_valid = True
     strings_to_check = [
         "_t1.nii.gz",
+        "_t1c.nii.gz",
         "_t1ce.nii.gz",
         "_t2.nii.gz",
         "_flair.nii.gz",