1 files changed, 2 insertions, 2 deletions
diff --git a/src/prog.c b/src/prog.c
index e7f16ab..46e3eef 100644
--- a/src/prog.c
+++ b/src/prog.c
@@ -119,7 +119,7 @@ int main(int argc, char** argv)
 	  //-PROCESS-TIFF-TO-LABELS------------------------
 	  //-----------------------------------------------
 	  ImageMask *file_im = tif_to_labels(fpath, &width, &height, &starting_label);
-	  ImageMaskData_t *file_labels = file_im->mask[0];
+	  ImageMaskData_t *file_labels = file_im->image[0];
 	  if (file_labels == NULL) {
 	    free(fpath);
 	    free(file_list[index]);
@@ -137,7 +137,7 @@ int main(int argc, char** argv)
       }
     }
   }
-  masks = masks_im->mask[0];
+  masks = masks_im->image[0];
   if (masks == NULL) {
     fprintf(stderr, "No masks found!\n");
     return 1;