diff --git a/src/Composer.php b/src/Composer.php
index d8f6669..63ffeae 100644
--- a/src/Composer.php
+++ b/src/Composer.php
@@ -32,11 +32,15 @@ public static function getModuleAliases(): Collection
         $packages = new Collection($data['packages'] ?? null);
 
         return $packages
-            ->whereIn('type', ['yii-module', 'craft-module'])
-            ->flatMap(function($package) {
+            ->flatMap(function(array $package) {
                 $packageName = $package['name'] ?? null;
+                $packageType = $package['type'] ?? null;
 
-                if (!$packageName) {
+                if (
+                    !$packageName ||
+                    !$packageType ||
+                    !preg_match('/^(craft|yii)/', $packageType)
+                ) {
                     return null;
                 }