diff --git a/mediapipe/tasks/ios/components/containers/sources/MPPCategory.h b/mediapipe/tasks/ios/components/containers/sources/MPPCategory.h
index bb5526dc4..7a760a52b 100644
--- a/mediapipe/tasks/ios/components/containers/sources/MPPCategory.h
+++ b/mediapipe/tasks/ios/components/containers/sources/MPPCategory.h
@@ -57,7 +57,7 @@ NS_SWIFT_NAME(ClassificationCategory)
 - (instancetype)initWithIndex:(NSInteger)index
                         score:(float)score
                  categoryName:(nullable NSString *)categoryName
-                  displayName:(nullable NSString *)displayName;
+                  displayName:(nullable NSString *)displayName NS_DESIGNATED_INITIALIZER;
 
 - (instancetype)init NS_UNAVAILABLE;