diff --git a/src/Commands/Concerns/TenantAware.php b/src/Commands/Concerns/TenantAware.php
index 3674be6..14c0340 100644
--- a/src/Commands/Concerns/TenantAware.php
+++ b/src/Commands/Concerns/TenantAware.php
@@ -4,19 +4,20 @@
 
 use Illuminate\Support\Arr;
 use Spatie\Multitenancy\Concerns\UsesMultitenancyConfig;
-use Spatie\Multitenancy\Models\Tenant;
+use Spatie\Multitenancy\Models\Concerns\UsesTenantModel;
 use Symfony\Component\Console\Input\InputInterface;
 use Symfony\Component\Console\Output\OutputInterface;
 
 trait TenantAware
 {
-    use UsesMultitenancyConfig;
+    use UsesMultitenancyConfig,
+        UsesTenantModel;
 
     protected function execute(InputInterface $input, OutputInterface $output)
     {
         $tenants = Arr::wrap($this->option('tenant'));
 
-        $tenantQuery = Tenant::query()
+        $tenantQuery = $this->getTenantModel()::query()
             ->when(! blank($tenants), function ($query) use ($tenants) {
                 collect($this->getTenantArtisanSearchFields())
                     ->each(fn ($field) => $query->orWhereIn($field, Arr::wrap($tenants)));