diff --git a/src/Processors/ExpandTraits.php b/src/Processors/ExpandTraits.php
index bdcdabcd..a4313a6c 100644
--- a/src/Processors/ExpandTraits.php
+++ b/src/Processors/ExpandTraits.php
@@ -44,7 +44,7 @@ public function __invoke(Analysis $analysis)
         }
 
         foreach ($schemas as $schema) {
-            if ($schema->_context->is('class') && !$schema->_aux) {
+            if ($schema->_context->is('class') && !$schema->_context->is('generated')) {
                 // look at class traits
                 $traits = $analysis->getTraitsOfClass($schema->_context->fullyQualifiedName($schema->_context->class), true);
                 $existing = [];