diff --git a/yudao-module-alert/yudao-module-alert-biz/src/main/java/cn/iocoder/yudao/module/alert/service/model/impl/ModelServiceImpl.java b/yudao-module-alert/yudao-module-alert-biz/src/main/java/cn/iocoder/yudao/module/alert/service/model/impl/ModelServiceImpl.java
index 9b38c51..b2abfb1 100644
--- a/yudao-module-alert/yudao-module-alert-biz/src/main/java/cn/iocoder/yudao/module/alert/service/model/impl/ModelServiceImpl.java
+++ b/yudao-module-alert/yudao-module-alert-biz/src/main/java/cn/iocoder/yudao/module/alert/service/model/impl/ModelServiceImpl.java
@@ -14,11 +14,7 @@ import cn.iocoder.yudao.module.alert.common.enums.ModelTrash;
 import cn.iocoder.yudao.module.alert.common.enums.ModelVisible;
 import cn.iocoder.yudao.module.alert.controller.admin.model.model.ModelInfo;
 import cn.iocoder.yudao.module.alert.controller.admin.model.vo.*;
-import cn.iocoder.yudao.module.alert.dao.domain.AssessReportCfg;
-import cn.iocoder.yudao.module.alert.dao.domain.ModelCfg;
-import cn.iocoder.yudao.module.alert.dao.domain.ModelVersion;
-import cn.iocoder.yudao.module.alert.dao.domain.SystemCfg;
-import cn.iocoder.yudao.module.alert.dao.domain.UnitCfg;
+import cn.iocoder.yudao.module.alert.dao.domain.*;
 import cn.iocoder.yudao.module.alert.dao.mapper.AssessReportCfgMapper;
 import cn.iocoder.yudao.module.alert.dao.service.ModelCfgService;
 import cn.iocoder.yudao.module.alert.dao.service.ModelVersionService;
@@ -75,10 +71,12 @@ public class ModelServiceImpl implements ModelService {
         Integer status = query.getStatus();
         Integer trash = query.getTrash();
         Integer typeId = query.getTypeId();
+        Integer unitId = query.getUnitId();
         List<Integer> systems = null;
         if (Objects.nonNull(typeId)) {
             LambdaQueryWrapper<SystemCfg> systemQuery = new LambdaQueryWrapper<>();
-            systemQuery.eq(SystemCfg::getSystemTypeId, typeId);
+            systemQuery.eq(SystemCfg::getSystemTypeId, typeId)
+                    .eq(SystemCfg::getUnitId, unitId);
             List<SystemCfg> list = systemCfgService.list(systemQuery);
             systems = list.stream().map(SystemCfg::getSystemId).collect(Collectors.toList());
         }