diff --git a/StubsController.php b/StubsController.php
index f063a0a..689b932 100644
--- a/StubsController.php
+++ b/StubsController.php
@@ -87,11 +87,11 @@ public function actionIndex()
                     $userIdentities[] = $component['identityClass'];
                 }
 
-                if (!isset($component['class'])) {
-                    continue;
+                if (isset($component['class'])) {
+                    $components[$name][] = $component['class'];
+                } elseif (isset($component['__class'])) {
+                    $components[$name][] = $component['__class'];
                 }
-
-                $components[$name][] = $component['class'];
             }
         }