diff --git a/tests/Behat/Behaviour/ContainsError.php b/tests/Behat/Behaviour/ContainsError.php
index d344afc..f097124 100644
--- a/tests/Behat/Behaviour/ContainsError.php
+++ b/tests/Behat/Behaviour/ContainsError.php
@@ -19,7 +19,7 @@ trait ContainsError
 {
     use DocumentAccessor;
 
-    public function hasError(string $message, string $errorClass = '.sylius-validation-error'): bool
+    public function hasError(string $message, string $errorClass = '.invalid-feedback'): bool
     {
         $errors = $this->getDocument()->findAll('css', $errorClass);
 
diff --git a/tests/Behat/Resources/services/contexts.yml b/tests/Behat/Resources/services/contexts.yml
index 4e92430..feb3abd 100644
--- a/tests/Behat/Resources/services/contexts.yml
+++ b/tests/Behat/Resources/services/contexts.yml
@@ -40,10 +40,10 @@ services:
             - '@bitbag.behat.page.admin.shipping_gateway.update'
             - '@sylius.behat.current_page_resolver'
             - '@sylius.behat.shared_storage'
-            - '@sylius.behat.notification_checker'
+            - '@sylius.behat.notification_checker.admin'
 
     bitbag.behat.context.ui.admin.shipping_export:
         class: Tests\BitBag\SyliusShippingExportPlugin\Behat\Context\Ui\Admin\ShippingExportContext
         arguments:
             - '@bitbag.behat.page.admin.shipping_export.index'
-            - '@sylius.behat.notification_checker'
+            - '@sylius.behat.notification_checker.admin'