diff --git a/src/DatabaseStateServiceProvider.php b/src/DatabaseStateServiceProvider.php
index 472ff9b..b21d6ce 100644
--- a/src/DatabaseStateServiceProvider.php
+++ b/src/DatabaseStateServiceProvider.php
@@ -2,7 +2,6 @@
 
 namespace pxlrbt\LaravelDatabaseState;
 
-use App\Console\Commands\SeedDatabaseStateCommand;
 use Illuminate\Console\Events\CommandFinished;
 use Illuminate\Support\Facades\Artisan;
 use Illuminate\Support\Facades\Event;
@@ -26,7 +25,7 @@ public function configurePackage(Package $package): void
                 Commands\SeedDatabaseStateCommand::class,
             ]);
 
-        if (class_exists('\Stancl\Tenancy\Concerns\HasTenantOptions')) {
+        if (trait_exists('\Stancl\Tenancy\Concerns\HasTenantOptions')) {
             $package->hasCommands([
                 Commands\SeedTenantsDatabaseStateCommand::class,
             ]);
@@ -45,7 +44,6 @@ public function packageBooted()
         );
     }
 
-
     public function runDatabaseStateSeeder(CommandFinished $event): void
     {
         if ($event->exitCode !== 0) {