diff --git a/sdks/python/apache_beam/ml/transforms/tft_test.py b/sdks/python/apache_beam/ml/transforms/tft_test.py
index f5615e9d4ad1..6afe9b5ab302 100644
--- a/sdks/python/apache_beam/ml/transforms/tft_test.py
+++ b/sdks/python/apache_beam/ml/transforms/tft_test.py
@@ -962,7 +962,7 @@ def validate_count_per_key(key_vocab_filename):
     self.assertEqual(expected_data, actual_data)
 
 
-class HashWordsTest(unittest.TestCase):
+class HashStringsTest(unittest.TestCase):
   def setUp(self) -> None:
     self.artifact_location = tempfile.mkdtemp()