diff --git a/src/contracts/Factory/IbexaRectorConfigFactory.php b/src/contracts/Factory/IbexaRectorConfigFactory.php
index ef6b656..e0553fb 100644
--- a/src/contracts/Factory/IbexaRectorConfigFactory.php
+++ b/src/contracts/Factory/IbexaRectorConfigFactory.php
@@ -62,6 +62,7 @@ public function createConfig(): RectorConfigBuilder
                    $this->extraSets
                )
            )
+           ->withAttributesSets(behat: true)
         ;
     }
 }
diff --git a/tests/contracts/IbexaRectorConfigFactoryTest.php b/tests/contracts/IbexaRectorConfigFactoryTest.php
index 3016408..d4b1761 100644
--- a/tests/contracts/IbexaRectorConfigFactoryTest.php
+++ b/tests/contracts/IbexaRectorConfigFactoryTest.php
@@ -14,6 +14,7 @@
 use Rector\Config\RectorConfig;
 use Rector\Configuration\Option;
 use Rector\Configuration\Parameter\SimpleParameterProvider;
+use Rector\Set\ValueObject\SetList;
 use Rector\Symfony\Set\SymfonySetList;
 
 /**
@@ -57,6 +58,7 @@ public static function getSetsForIbexaConfigFactory(): iterable
             SymfonySetList::SYMFONY_70,
             SymfonySetList::SYMFONY_71,
             SymfonySetList::SYMFONY_72,
+            SetList::BEHAT_ANNOTATIONS_TO_ATTRIBUTES,
         ];
 
         yield 'default set list' => [