diff --git a/test/test_flaky_decorator.py b/test/test_flaky_decorator.py
index b037332..d17c4fc 100644
--- a/test/test_flaky_decorator.py
+++ b/test/test_flaky_decorator.py
@@ -38,13 +38,13 @@ def test_something():
         }
 
         self.assertIsNotNone(flaky_attribute)
-        self.assertDictContainsSubset(
+        self.assertLessThanOrEqualTo(
             {
                 FlakyNames.MIN_PASSES: min_passes,
                 FlakyNames.MAX_RUNS: max_runs,
                 FlakyNames.CURRENT_PASSES: 0,
                 FlakyNames.CURRENT_RUNS: 0,
                 FlakyNames.CURRENT_ERRORS: None
-            },
-            flaky_attribute
+            }.items(),
+            flaky_attribute.items()
         )