diff --git a/tests/test_base.py b/tests/test_base.py
index 0e9d667d..dc44a499 100644
--- a/tests/test_base.py
+++ b/tests/test_base.py
@@ -5,7 +5,7 @@
 from awacs.aws import Action, PolicyDocument
 
 
-class AWSHelperFn(object):
+class AWSHelperFn:
     """Mock class to test classes with non-awacs AWSHelperFn parent."""
 
     def to_json(self, indent=4, sort_keys=True):
@@ -53,7 +53,7 @@ def test_invalid_property(self):
 
 class TestAWSProperty(unittest.TestCase):
     def test_prop_value_type_mismatch_expect_list(self):
-        class InvalidClass(object):
+        class InvalidClass:
             """Class of invalid type."""
 
         tests_values = ["val", {"key": "val"}, {"val"}, tuple("val"), InvalidClass()]