diff --git a/tests/Behat/Context/Api/PagolightContext.php b/tests/Behat/Context/Api/PagolightContext.php
index 0c0ba69..a535b9b 100644
--- a/tests/Behat/Context/Api/PagolightContext.php
+++ b/tests/Behat/Context/Api/PagolightContext.php
@@ -4,13 +4,16 @@
 
 namespace Tests\Webgriffe\SyliusPagolightPlugin\Behat\Context\Api;
 
+if (!interface_exists(\Sylius\Resource\Doctrine\Persistence\RepositoryInterface::class)) {
+    class_alias(\Sylius\Component\Resource\Repository\RepositoryInterface::class, \Sylius\Resource\Doctrine\Persistence\RepositoryInterface::class);
+}
 use Behat\Behat\Context\Context;
 use GuzzleHttp\Client;
 use GuzzleHttp\Psr7\Request;
 use Psr\Http\Client\ClientInterface;
 use Sylius\Bundle\PayumBundle\Model\PaymentSecurityTokenInterface;
 use Sylius\Component\Core\Repository\PaymentRepositoryInterface;
-use Sylius\Component\Resource\Repository\RepositoryInterface;
+use Sylius\Resource\Doctrine\Persistence\RepositoryInterface;
 use Symfony\Component\Routing\Generator\UrlGeneratorInterface;
 use Symfony\Component\Routing\RequestContext;
 use Tests\Webgriffe\SyliusPagolightPlugin\Behat\Context\PayumPaymentTrait;
diff --git a/tests/Behat/Context/PayumPaymentTrait.php b/tests/Behat/Context/PayumPaymentTrait.php
index b8dd87d..0348239 100644
--- a/tests/Behat/Context/PayumPaymentTrait.php
+++ b/tests/Behat/Context/PayumPaymentTrait.php
@@ -7,7 +7,7 @@
 use Sylius\Bundle\PayumBundle\Model\PaymentSecurityTokenInterface;
 use Sylius\Component\Core\Model\PaymentInterface;
 use Sylius\Component\Core\Repository\PaymentRepositoryInterface;
-use Sylius\Component\Resource\Repository\RepositoryInterface;
+use Sylius\Resource\Doctrine\Persistence\RepositoryInterface;
 use Webmozart\Assert\Assert;
 
 trait PayumPaymentTrait
diff --git a/tests/Behat/Context/Ui/PagolightContext.php b/tests/Behat/Context/Ui/PagolightContext.php
index 4da6f5e..79f1eee 100644
--- a/tests/Behat/Context/Ui/PagolightContext.php
+++ b/tests/Behat/Context/Ui/PagolightContext.php
@@ -4,6 +4,9 @@
 
 namespace Tests\Webgriffe\SyliusPagolightPlugin\Behat\Context\Ui;
 
+if (!interface_exists(\Sylius\Resource\Doctrine\Persistence\RepositoryInterface::class)) {
+    class_alias(\Sylius\Component\Resource\Repository\RepositoryInterface::class, \Sylius\Resource\Doctrine\Persistence\RepositoryInterface::class);
+}
 use Behat\Behat\Context\Context;
 use Behat\Mink\Session;
 use Sylius\Behat\Page\Shop\Order\ShowPageInterface;
@@ -12,7 +15,7 @@
 use Sylius\Component\Core\Model\OrderInterface;
 use Sylius\Component\Core\Repository\OrderRepositoryInterface;
 use Sylius\Component\Core\Repository\PaymentRepositoryInterface;
-use Sylius\Component\Resource\Repository\RepositoryInterface;
+use Sylius\Resource\Doctrine\Persistence\RepositoryInterface;
 use Symfony\Component\Routing\Generator\UrlGeneratorInterface;
 use Symfony\Component\Routing\RequestContext;
 use Tests\Webgriffe\SyliusPagolightPlugin\Behat\Context\PayumPaymentTrait;