diff -u b/tests/src/Unit/CurrentUserContextTest.php b/tests/src/Unit/CurrentUserContextTest.php
--- b/tests/src/Unit/CurrentUserContextTest.php
+++ b/tests/src/Unit/CurrentUserContextTest.php
@@ -3,9 +3,9 @@
 namespace Drupal\Tests\page_manager\Unit;
 
 use Drupal\Core\Plugin\Context\Context;
-use Drupal\Core\Plugin\Context\ContextRepositoryInterface;
 use Drupal\Core\Plugin\Context\EntityContext;
 use Drupal\Core\Plugin\Context\EntityContextDefinition;
+use Drupal\Core\Plugin\Context\LazyContextRepository;
 use Drupal\Core\Session\AccountProxyInterface;
 use Drupal\page_manager\EventSubscriber\CurrentUserContext;
 use Prophecy\Argument;
@@ -27,7 +27,7 @@
       ->disableOriginalConstructor()
       ->getMock();
 
-    $contextRepository = $this->getMockBuilder(ContextRepositoryInterface::class)
+    $contextRepository = $this->getMockBuilder(LazyContextRepository::class)
       ->disableOriginalConstructor()
       ->getMock();
     $currentUserContext = new EntityContext(new EntityContextDefinition('user', 'current_user_context'), $currentUser->getAccount());