diff --git a/composer.json b/composer.json
index 2a3cb2df..64ae28a5 100644
--- a/composer.json
+++ b/composer.json
@@ -32,7 +32,7 @@
     },
     "require-dev": {
         "guzzlehttp/guzzle": "^6.5||^7.4",
-        "phpspec/prophecy-phpunit": "^1.1",
+        "phpspec/prophecy-phpunit": "^2.0",
         "phpunit/phpunit": "^7.5||^9.5",
         "psr/cache": "^1.0||^2.0",
         "psr/http-client": "^1.0",
diff --git a/tests/CachedKeySetTest.php b/tests/CachedKeySetTest.php
index 1dc2fdc0..7fec41b3 100644
--- a/tests/CachedKeySetTest.php
+++ b/tests/CachedKeySetTest.php
@@ -6,6 +6,7 @@
 use OutOfBoundsException;
 use PHPUnit\Framework\TestCase;
 use Prophecy\Argument;
+use Prophecy\PhpUnit\ProphecyTrait;
 use Psr\Cache\CacheItemInterface;
 use Psr\Cache\CacheItemPoolInterface;
 use Psr\Http\Client\ClientInterface;
@@ -14,6 +15,8 @@
 
 class CachedKeySetTest extends TestCase
 {
+    use ProphecyTrait;
+
     private $testJwksUri = 'https://jwk.uri';
     private $testJwksUriKey = 'jwkshttpsjwk.uri';
     private $testJwks1 = '{"keys": [{"kid":"foo","kty":"RSA","alg":"foo","n":"","e":""}]}';