diff --git a/tests/test_client.py b/tests/test_client.py
index b180c6b8..1342d29e 100644
--- a/tests/test_client.py
+++ b/tests/test_client.py
@@ -190,11 +190,16 @@ def test_auth_code_flow(self):
             self.assertLoosely(result, lambda: self.assertIn('access_token', result))
 
     def test_auth_code_flow_error_response(self):
-        with self.assertRaisesRegexp(ValueError, "state missing"):
+        if hasattr(self, "assertRaisesRegex"):
+            _assert = self.assertRaisesRegex
+        else:
+            _assert = self.assertRaisesRegexp
+
+        with _assert(ValueError, "state missing"):
             self.client.obtain_token_by_auth_code_flow({}, {"code": "foo"})
-        with self.assertRaisesRegexp(ValueError, "state mismatch"):
+        with _assert(ValueError, "state mismatch"):
             self.client.obtain_token_by_auth_code_flow({"state": "1"}, {"state": "2"})
-        with self.assertRaisesRegexp(ValueError, "scope"):
+        with _assert(ValueError, "scope"):
             self.client.obtain_token_by_auth_code_flow(
                 {"state": "s", "scope": ["foo"]}, {"state": "s"}, scope=["bar"])
         self.assertEqual(