diff --git a/tests/src/Kernel/ServiceAccessControlHandlerTest.php b/tests/src/Kernel/ServiceAccessControlHandlerTest.php
index bd2c8f7..cba8d8e 100644
--- a/tests/src/Kernel/ServiceAccessControlHandlerTest.php
+++ b/tests/src/Kernel/ServiceAccessControlHandlerTest.php
@@ -2,6 +2,7 @@
 
 namespace Drupal\Tests\transbank\Kernel;
 
+use Prophecy\PhpUnit\ProphecyTrait;
 use Drupal\Core\Access\AccessResult;
 use Drupal\Core\Cache\Context\CacheContextsManager;
 use Drupal\Core\DependencyInjection\ContainerBuilder;
@@ -15,6 +16,7 @@ use Drupal\transbank\Entity\Service;
  */
 class ServiceAccessControlHandlerTest extends KernelTestBase {
 
+  use ProphecyTrait;
   use UserCreationTrait;
 
   /**
diff --git a/tests/src/Unit/ServiceTest.php b/tests/src/Unit/ServiceTest.php
index d23cdaf..9b64931 100644
--- a/tests/src/Unit/ServiceTest.php
+++ b/tests/src/Unit/ServiceTest.php
@@ -2,6 +2,7 @@
 
 namespace Drupal\Tests\transbank\Unit;
 
+use Prophecy\PhpUnit\ProphecyTrait;
 use Drupal\Core\DependencyInjection\ContainerBuilder;
 use Drupal\Tests\UnitTestCase;
 use Drupal\transbank\Entity\Service;
@@ -17,6 +18,7 @@ use Prophecy\Argument;
  */
 class ServiceTest extends UnitTestCase {
 
+  use ProphecyTrait;
   /**
    * {@inheritdoc}
    */