diff --git a/tests/src/Unit/ConstraintsTest.php b/tests/src/Unit/ConstraintsTest.php
index fb1edd6..c6a58cc 100644
--- a/tests/src/Unit/ConstraintsTest.php
+++ b/tests/src/Unit/ConstraintsTest.php
@@ -25,9 +25,7 @@ class ConstraintsTest extends UnitTestCase {
     $this->assertEquals('At least one slideshow item must exist.', $constraint->message, 'Correct constraint message found.');
 
     // Test the validator with valid values.
-    $execution_context = $this->getMockBuilder('\Drupal\Core\TypedData\Validation\ExecutionContext')
-      ->disableOriginalConstructor()
-      ->getMock();
+    $execution_context = $this->createMock('\Drupal\Core\TypedData\Validation\ExecutionContext');
 
     $execution_context->expects($this->exactly(0))
       ->method('addViolation');
@@ -39,9 +37,7 @@ class ConstraintsTest extends UnitTestCase {
     $validator->validate($value, $constraint);
 
     // Test the validator with invalid values.
-    $execution_context = $this->getMockBuilder('\Drupal\Core\TypedData\Validation\ExecutionContext')
-      ->disableOriginalConstructor()
-      ->getMock();
+    $execution_context = $this->createMock('\Drupal\Core\TypedData\Validation\ExecutionContext');
 
     $execution_context->expects($this->exactly(1))
       ->method('addViolation')