diff --git a/tests/src/Unit/ReplicatorTest.php b/tests/src/Unit/ReplicatorTest.php
index 0daf3ae..aebd232 100644
--- a/tests/src/Unit/ReplicatorTest.php
+++ b/tests/src/Unit/ReplicatorTest.php
@@ -2,6 +2,7 @@
 
 namespace Drupal\Tests\replicate\Unit;
 
+use Prophecy\PhpUnit\ProphecyTrait;
 use Drupal\Core\Entity\EntityInterface;
 use Drupal\Core\Entity\EntityTypeManagerInterface;
 use Drupal\Core\Entity\FieldableEntityInterface;
@@ -23,6 +24,7 @@ use Symfony\Component\EventDispatcher\EventDispatcherInterface;
  */
 class ReplicatorTest extends UnitTestCase {
 
+  use ProphecyTrait;
   /**
    * Tests the cloneEntity method.
    *