diff --git a/composer.json b/composer.json
index 0dcf826f..682a014d 100644
--- a/composer.json
+++ b/composer.json
@@ -59,6 +59,14 @@
         },
         "files": ["tests/bootstrap.php"]
     },
+    "extra": {
+        "config-plugin-options": {
+            "source-directory": "config"
+        },
+        "config-plugin": {
+            "di": "di.php"
+        }
+    },
     "config": {
         "sort-packages": true,
         "allow-plugins": {
diff --git a/tests/ConfigTest.php b/tests/ConfigTest.php
index f79912c4..0d7b3e8e 100644
--- a/tests/ConfigTest.php
+++ b/tests/ConfigTest.php
@@ -12,8 +12,6 @@
 use Yiisoft\Rbac\ItemsStorageInterface;
 use Yiisoft\Rbac\Manager;
 use Yiisoft\Rbac\ManagerInterface;
-use Yiisoft\Rbac\RuleFactoryInterface;
-use Yiisoft\Rbac\SimpleRuleFactory;
 use Yiisoft\Rbac\Tests\Support\FakeAssignmentsStorage;
 use Yiisoft\Rbac\Tests\Support\FakeItemsStorage;
 
@@ -33,7 +31,6 @@ private function createContainer(): ContainerInterface
         $definitions = array_merge($definitions, [
             ItemsStorageInterface::class => FakeItemsStorage::class,
             AssignmentsStorageInterface::class => FakeAssignmentsStorage::class,
-            RuleFactoryInterface::class => SimpleRuleFactory::class,
         ]);
         $config = ContainerConfig::create()->withDefinitions($definitions);