diff --git a/src/StorageDrivers/Database/DomainModel.php b/src/StorageDrivers/Database/DomainModel.php
index b52dff8a..17c88ca1 100644
--- a/src/StorageDrivers/Database/DomainModel.php
+++ b/src/StorageDrivers/Database/DomainModel.php
@@ -5,6 +5,7 @@ declare(strict_types=1);
 namespace Stancl\Tenancy\StorageDrivers\Database;
 
 use Illuminate\Database\Eloquent\Model;
+use Stancl\Tenancy\DatabaseManager;
 
 /**
  * @internal Class is subject to breaking changes in minor and patch versions.
diff --git a/src/StorageDrivers/Database/TenantModel.php b/src/StorageDrivers/Database/TenantModel.php
index 80b3b623..f0819b98 100644
--- a/src/StorageDrivers/Database/TenantModel.php
+++ b/src/StorageDrivers/Database/TenantModel.php
@@ -5,6 +5,7 @@ declare(strict_types=1);
 namespace Stancl\Tenancy\StorageDrivers\Database;
 
 use Illuminate\Database\Eloquent\Model;
+use Stancl\Tenancy\DatabaseManager;
 
 /**
  * @internal Class is subject to breaking changes in minor and patch versions.