diff --git a/mediapipe/tasks/ios/vision/core/sources/MPPImage.h b/mediapipe/tasks/ios/vision/core/sources/MPPImage.h
index f796c76c9..44f5f9ba9 100644
--- a/mediapipe/tasks/ios/vision/core/sources/MPPImage.h
+++ b/mediapipe/tasks/ios/vision/core/sources/MPPImage.h
@@ -19,7 +19,7 @@
 NS_ASSUME_NONNULL_BEGIN
 
 /** Types of image sources. */
-typedef NSInteger MPPTaskImageSourceType NS_TYPED_ENUM NS_SWIFT_NAME(MLImageSourceType);
+typedef NSInteger MPPTaskImageSourceType NS_TYPED_ENUM NS_SWIFT_NAME(TaskImageSourceType);
 /** Image source is a `UIImage`. */
 static const MPPTaskImageSourceType MPPTaskImageSourceTypeImage = 0;
 /** Image source is a `CVPixelBuffer`. */