1 files changed, 2 insertions, 2 deletions
diff --git a/src/prog.c b/src/prog.c
index 87b7bb3..3335268 100644
--- a/src/prog.c
+++ b/src/prog.c
@@ -88,7 +88,7 @@ int main(int argc, char** argv)
   uint32_t width, height;
   MaskData_t starting_label = 1;
   MaskData_t *masks = NULL;
-  ImageMask *masks_im = NULL;
+  Mask *masks_im = NULL;
   // Expect a directory to be passed as the first argument
   if (directory != NULL) {
     // Ensure the directory exists
@@ -118,7 +118,7 @@ int main(int argc, char** argv)
 	  //-----------------------------------------------
 	  //-PROCESS-TIFF-TO-LABELS------------------------
 	  //-----------------------------------------------
-	  ImageMask *file_im = tif_to_labels(fpath, &width, &height, &starting_label);
+	  Mask *file_im = tif_to_labels(fpath, &width, &height, &starting_label);
 	  MaskData_t *file_labels = file_im->image[0];
 	  if (file_labels == NULL) {
 	    free(fpath);