diff --git a/ci/deptrac/config.yml b/ci/deptrac/config.yml
index 4c231735..3b368901 100644
--- a/ci/deptrac/config.yml
+++ b/ci/deptrac/config.yml
@@ -6,20 +6,20 @@ deptrac:
   layers:
     - name: Functions
       collectors:
-        - type: className
-          regex: \\MartinGeorgiev\\Doctrine\\ORM\\AST\\Functions\\.*
+        - type: class
+          value: \\MartinGeorgiev\\Doctrine\\ORM\\AST\\Functions\\.*
     - name: Types
       collectors:
-        - type: className
-          regex: \\MartinGeorgiev\\Doctrine\\DBAL\\Types\\.*
+        - type: class
+          value: \\MartinGeorgiev\\Doctrine\\DBAL\\Types\\.*
     - name: Utils
       collectors:
-        - type: className
-          regex: \\MartinGeorgiev\\Utils\\.*
+        - type: class
+          value: \\MartinGeorgiev\\Utils\\.*
     - name: Tests
       collectors:
-        - type: className
-          regex: \\Tests\\MartinGeorgiev\\.*
+        - type: class
+          value: \\Tests\\MartinGeorgiev\\.*
 
   ruleset:
     Functions:
diff --git a/composer.json b/composer.json
index a4f9c3e8..f5aecde6 100644
--- a/composer.json
+++ b/composer.json
@@ -50,7 +50,7 @@
         "phpstan/phpstan": "^1.10.65",
         "phpstan/phpstan-phpunit": "^1.3.16",
         "phpunit/phpunit": "^10.5.15",
-        "qossmic/deptrac-shim": "^1.0.2",
+        "qossmic/deptrac": "^2.0.0-alpha",
         "rector/rector": "^1.0.3",
         "symfony/cache": "^6.4||^7.0"
     },