diff --git a/mediapipe/tasks/ios/vision/image_segmenter/sources/MPPImageSegmenterOptions.m b/mediapipe/tasks/ios/vision/image_segmenter/sources/MPPImageSegmenterOptions.m
index 282a729bb..8dd58357f 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.shouldOutputConfidenceMasks;
+  imageSegmenterOptions.shouldOutputCategoryMasks = self.shouldOutputCategoryMasks;
   imageSegmenterOptions.displayNamesLocale = self.displayNamesLocale;
   imageSegmenterOptions.imageSegmenterLiveStreamDelegate = self.imageSegmenterLiveStreamDelegate;