diff --git a/tensorflow/python/keras/saving/saved_model/saved_model_test.py b/tensorflow/python/keras/saving/saved_model/saved_model_test.py
index b26ce9f8b23..d685a649f5b 100644
--- a/tensorflow/python/keras/saving/saved_model/saved_model_test.py
+++ b/tensorflow/python/keras/saving/saved_model/saved_model_test.py
@@ -1161,7 +1161,7 @@ class MetricTest(test.TestCase, parameterized.TestCase):
 
     saved_model_dir = self._save_model_dir()
     tf_save.save(model, saved_model_dir)
-    with self.assertRaisesRegex(ValueError, 'custom_objects'):
+    with self.assertRaisesRegex(ValueError, 'metric'):
       keras_load.load(saved_model_dir)
 
     keras_load.load(saved_model_dir, compile=False)