diff --git a/src/Barryvdh/LaravelIdeHelper/GeneratorCommand.php b/src/Barryvdh/LaravelIdeHelper/GeneratorCommand.php
index 2340847..e1e1d55 100755
--- a/src/Barryvdh/LaravelIdeHelper/GeneratorCommand.php
+++ b/src/Barryvdh/LaravelIdeHelper/GeneratorCommand.php
@@ -138,8 +138,8 @@ class GeneratorCommand extends Command {
         $aliases = $aliasLoader->getAliases();
 
         foreach($aliases as $alias => $facade){
-            //Check if the facade is not a Trait (could need a better check..)
-            if(\Str::contains($facade, 'Trait')){
+            //Check if the facade is not a Trait
+            if(function_exists('trait_exists') && trait_exists($facade)){
                 continue;
             }
             $facade = '\\'.ltrim($facade, '\\');