diff --git a/server/scripts/setup-database.js b/server/scripts/setup-database.js
index ceb4dbc..3989de2 100644
--- a/server/scripts/setup-database.js
+++ b/server/scripts/setup-database.js
@@ -1,34 +1,22 @@
 const { Sequelize } = require('sequelize');
 const bcrypt = require('bcryptjs');
 
-// Import models
-const Device = require('../models/Device');
-const DroneDetection = require('../models/DroneDetection');
-const Heartbeat = require('../models/Heartbeat');
-const User = require('../models/User');
-const AlertRule = require('../models/AlertRule');
-const AlertLog = require('../models/AlertLog');
+// Import models from the main models index
+const {
+  sequelize,
+  Device,
+  DroneDetection,
+  Heartbeat,
+  User,
+  AlertRule,
+  AlertLog,
+  Tenant
+} = require('../models');
 
 const setupDatabase = async () => {
   try {
     console.log('🚀 Starting database setup...\n');
 
-    // Load environment variables
-    require('dotenv').config();
-
-    // Create Sequelize instance
-    const sequelize = new Sequelize(
-      process.env.DB_NAME,
-      process.env.DB_USER,
-      process.env.DB_PASSWORD,
-      {
-        host: process.env.DB_HOST,
-        port: process.env.DB_PORT,
-        dialect: 'postgres',
-        logging: console.log,
-      }
-    );
-
     // Test database connection
     console.log('📡 Testing database connection...');
     await sequelize.authenticate();