diff --git a/src/Console/ModelsCommand.php b/src/Console/ModelsCommand.php
index a56865d..5abc256 100644
--- a/src/Console/ModelsCommand.php
+++ b/src/Console/ModelsCommand.php
@@ -303,7 +303,7 @@ class ModelsCommand extends Command
                         $args = $this->getParameters($reflection);
                         //Remove the first ($query) argument
                         array_shift($args);
-                        $this->setMethod($name, '\\' . $reflection->class, $args);
+                        $this->setMethod($name, '\Illuminate\Database\Query\Builder|\\' . $reflection->class, $args);
                     }
                 } elseif (!method_exists('Eloquent', $method) && !Str::startsWith($method, 'get')) {