diff --git a/StubsController.php b/StubsController.php
index 3063af8..e054ee4 100644
--- a/StubsController.php
+++ b/StubsController.php
@@ -91,7 +91,7 @@ public function actionIndex()
                 if (isset($component['class'])) {
                     $class = $component['class'];
                 } elseif (isset($component['__class'])) {
-                    $class = $component['class'];
+                    $class = $component['__class'];
                 } else {
                     continue;
                 }