diff --git a/tests/extensions/Retryable.php b/tests/extensions/Retryable.php
index b1b2f1862..61fec0ef1 100644
--- a/tests/extensions/Retryable.php
+++ b/tests/extensions/Retryable.php
@@ -52,7 +52,7 @@ trait Retryable
         $attributes = $method->getAttributes(Retry::class);
         $attribute = $attributes[0] ?? null;
         $args = $attribute?->getArguments();
-        $retries = $args['retries'] ?? 0;
+        $retries = $args['count'] ?? 0;
         return \max(0, $retries);
     }