diff --git a/server/models/AlertRule.js b/server/models/AlertRule.js
index c28c2dc..905bd0d 100644
--- a/server/models/AlertRule.js
+++ b/server/models/AlertRule.js
@@ -15,6 +15,14 @@ module.exports = (sequelize) => {
         key: 'id'
       }
     },
+    tenant_id: {
+      type: DataTypes.UUID,
+      allowNull: true, // Allow null for testing
+      references: {
+        model: 'tenants',
+        key: 'id'
+      }
+    },
     name: {
       type: DataTypes.STRING,
       allowNull: false,
diff --git a/server/tests/setup.js b/server/tests/setup.js
index 38b70e5..983a90d 100644
--- a/server/tests/setup.js
+++ b/server/tests/setup.js
@@ -68,6 +68,9 @@ async function setupTestEnvironment() {
   Tenant.hasMany(Device, { foreignKey: 'tenant_id', as: 'devices' });
   Device.belongsTo(Tenant, { foreignKey: 'tenant_id', as: 'tenant' });
 
+  Tenant.hasMany(AlertRule, { foreignKey: 'tenant_id', as: 'alertRules' });
+  AlertRule.belongsTo(Tenant, { foreignKey: 'tenant_id', as: 'tenant' });
+
   // Create models object
   models = {
     sequelize,