diff --git a/mediapipe/tasks/python/test/text/text_classifier_test.py b/mediapipe/tasks/python/test/text/text_classifier_test.py
index e864cc02b..8678d2194 100644
--- a/mediapipe/tasks/python/test/text/text_classifier_test.py
+++ b/mediapipe/tasks/python/test/text/text_classifier_test.py
@@ -125,7 +125,7 @@ class ModelFileType(enum.Enum):
   FILE_NAME = 2
 
 
-class ImageClassifierTest(parameterized.TestCase):
+class TextClassifierTest(parameterized.TestCase):
 
   def setUp(self):
     super().setUp()