diff --git a/tests/serializers/tests.py b/tests/serializers/tests.py
index 987a887f4b..ba5bb34434 100644
--- a/tests/serializers/tests.py
+++ b/tests/serializers/tests.py
@@ -487,8 +487,8 @@ class NoYamlSerializerTestCase(TestCase):
 
     def test_dumpdata_pyyaml_error_message(self):
         """Calling dumpdata produces an error when yaml package missing"""
-        self.assertRaisesRegexp(management.CommandError, YAML_IMPORT_ERROR_MESSAGE,
-                management.call_command, 'dumpdata', format='yaml')
+        with six.assertRaisesRegex(self, management.CommandError, YAML_IMPORT_ERROR_MESSAGE):
+            management.call_command('dumpdata', format='yaml')
 
 
 @unittest.skipUnless(HAS_YAML, "No yaml library detected")