diff --git a/mediapipe/tasks/python/test/text/text_embedder_test.py b/mediapipe/tasks/python/test/text/text_embedder_test.py
index b39f51a8d..8ef616150 100644
--- a/mediapipe/tasks/python/test/text/text_embedder_test.py
+++ b/mediapipe/tasks/python/test/text/text_embedder_test.py
@@ -50,7 +50,7 @@ class ModelFileType(enum.Enum):
   FILE_NAME = 2
 
 
-class ImageEmbedderTest(parameterized.TestCase):
+class TextEmbedderTest(parameterized.TestCase):
 
   def setUp(self):
     super().setUp()