diff --git a/tensorflow/python/module/module_test.py b/tensorflow/python/module/module_test.py
index 2d1b1627655..9b3816eefc6 100644
--- a/tensorflow/python/module/module_test.py
+++ b/tensorflow/python/module/module_test.py
@@ -335,8 +335,7 @@ class ForwardMethodsTest(test_util.TensorFlowTestCase):
 class AbcTest(test_util.TensorFlowTestCase):
 
   def testAbstract(self):
-    msg = "Can't instantiate .* abstract methods"
-    with self.assertRaisesRegex(TypeError, msg):
+    with self.assertRaises(TypeError):
       AbstractModule()  # pylint: disable=abstract-class-instantiated
 
   def testConcrete(self):