diff --git a/mediapipe/model_maker/python/vision/image_classifier/model_spec.py b/mediapipe/model_maker/python/vision/image_classifier/model_spec.py
index ef44f86e6..d46cafe6b 100644
--- a/mediapipe/model_maker/python/vision/image_classifier/model_spec.py
+++ b/mediapipe/model_maker/python/vision/image_classifier/model_spec.py
@@ -28,7 +28,7 @@ class ModelSpec(object):
                uri: str,
                input_image_shape: Optional[List[int]] = None,
                name: str = ''):
-    """Initializes a new instance of the `ImageModelSpec` class.
+    """Initializes a new instance of the image classifier `ModelSpec` class.
 
     Args:
       uri: str, URI to the pretrained model.