diff --git a/tests/src/BlockDisplayVariantTest.php b/tests/src/BlockDisplayVariantTest.php
index 2af448b..b951e2c 100644
--- a/tests/src/BlockDisplayVariantTest.php
+++ b/tests/src/BlockDisplayVariantTest.php
@@ -108,8 +108,11 @@ public function testBuild() {
     $account = $this->getMock('Drupal\Core\Session\AccountInterface');
     $uuid_generator = $this->getMock('Drupal\Component\Uuid\UuidInterface');
     $page_title = 'Page title';
+    $token = $this->getMockBuilder('Drupal\Core\Utility\Token')
+      ->disableOriginalConstructor()
+      ->getMock();
     $display_variant = $this->getMockBuilder('Drupal\page_manager\Plugin\DisplayVariant\BlockDisplayVariant')
-      ->setConstructorArgs(array(array('page_title' => $page_title), 'test', array(), $context_handler, $account, $uuid_generator))
+      ->setConstructorArgs(array(array('page_title' => $page_title), 'test', array(), $context_handler, $account, $uuid_generator, $token))
       ->setMethods(array('getBlockBag', 'drupalHtmlClass', 'renderPageTitle'))
       ->getMock();
     $display_variant->expects($this->exactly(1))