diff --git a/src/ingest_validation_tests/segmentation_mask_imagesize_validation.py b/src/ingest_validation_tests/segmentation_mask_imagesize_validation.py
index 6aba47c..ac818d2 100644
--- a/src/ingest_validation_tests/segmentation_mask_imagesize_validation.py
+++ b/src/ingest_validation_tests/segmentation_mask_imagesize_validation.py
@@ -65,7 +65,7 @@ def collect_errors(self, **kwargs) -> List[Optional[str]]:
                     "**/*.OME.TIFF",
                     "**/*.OME.TIF",
                 ]:
-                    for file in Path(GetParentData(row["parent_sample_id"],
+                    for file in Path(GetParentData(row["parent_dataset_id"],
                                                    self.globus_token,
                                                    self.app_context).get_path()).glob(glob_expr):
                         parent_filenames_to_test.append(file)