diff --git a/yudao-module-farm/src/main/java/cn/iocoder/yudao/module/farm/service/modelmanagement/ModelManagementServiceImpl.java b/yudao-module-farm/src/main/java/cn/iocoder/yudao/module/farm/service/modelmanagement/ModelManagementServiceImpl.java
index b2b5e08d..edd9514d 100644
--- a/yudao-module-farm/src/main/java/cn/iocoder/yudao/module/farm/service/modelmanagement/ModelManagementServiceImpl.java
+++ b/yudao-module-farm/src/main/java/cn/iocoder/yudao/module/farm/service/modelmanagement/ModelManagementServiceImpl.java
@@ -279,7 +279,7 @@ public class ModelManagementServiceImpl implements ModelManagementService {
         Map<String, String> unit2Map = unit2List.stream().collect(Collectors.toMap(SysDictDO::getField1Code, SysDictDO::getField1Name, (v1, v2) -> v1));
 
 
-        ModelManagementDO modelManagementDO = modelManagementMapper.selectById(id);
+        ModelManagementDO modelManagementDO = modelManagementMapper.selectOne(ModelManagementDO::getTaskId,id);
         if (modelManagementDO == null) {
             return null;
         }