diff --git a/mediapipe/tasks/ios/vision/image_segmenter/sources/MPPImageSegmenterOptions.h b/mediapipe/tasks/ios/vision/image_segmenter/sources/MPPImageSegmenterOptions.h
index 71f65253..b089ac7d 100644
--- a/mediapipe/tasks/ios/vision/image_segmenter/sources/MPPImageSegmenterOptions.h
+++ b/mediapipe/tasks/ios/vision/image_segmenter/sources/MPPImageSegmenterOptions.h
@@ -92,7 +92,7 @@ NS_SWIFT_NAME(ImageSegmenterOptions)
 @property(nonatomic) BOOL shouldOutputConfidenceMasks;
 
 /** Represents whether to output category mask. */
-@property(nonatomic) BOOL shouldOutputCategoryMasks;
+@property(nonatomic) BOOL shouldOutputCategoryMask;
 
 @end
 
diff --git a/mediapipe/tasks/ios/vision/image_segmenter/sources/MPPImageSegmenterOptions.m b/mediapipe/tasks/ios/vision/image_segmenter/sources/MPPImageSegmenterOptions.m
index 8dd58357..609d64e2 100644
--- a/mediapipe/tasks/ios/vision/image_segmenter/sources/MPPImageSegmenterOptions.m
+++ b/mediapipe/tasks/ios/vision/image_segmenter/sources/MPPImageSegmenterOptions.m
@@ -30,7 +30,7 @@
 
   imageSegmenterOptions.runningMode = self.runningMode;
   imageSegmenterOptions.shouldOutputConfidenceMasks = self.shouldOutputConfidenceMasks;
-  imageSegmenterOptions.shouldOutputCategoryMasks = self.shouldOutputCategoryMasks;
+  imageSegmenterOptions.shouldOutputCategoryMask = self.shouldOutputCategoryMask;
   imageSegmenterOptions.displayNamesLocale = self.displayNamesLocale;
   imageSegmenterOptions.imageSegmenterLiveStreamDelegate = self.imageSegmenterLiveStreamDelegate;