diff --git a/src/main/java/com/spring/modules/part/service/impl/BomManagementServiceImpl.java b/src/main/java/com/spring/modules/part/service/impl/BomManagementServiceImpl.java
index d08ec710..45617a56 100644
--- a/src/main/java/com/spring/modules/part/service/impl/BomManagementServiceImpl.java
+++ b/src/main/java/com/spring/modules/part/service/impl/BomManagementServiceImpl.java
@@ -633,6 +633,7 @@ public class BomManagementServiceImpl extends ServiceImpl<BomManagementMapper, B
         if (partInformationEntity == null) {
             throw new RuntimeException("子物料不存在!");
         }
+        data.setPrintUnit(partInformationEntity.getUmId());
 //        // 校验工序
 //        if (!data.getOperationId().isEmpty()) {
 //            OperationEntity operationEntity = new OperationEntity();
@@ -772,6 +773,11 @@ public class BomManagementServiceImpl extends ServiceImpl<BomManagementMapper, B
         if ("byProduct".equals(data.getProductFlag())) {
             data.setQtyPerAssembly(data.getQtyPerAssembly().negate());
         }
+        PartInformationEntity partInformationEntity = partInformationMapper.selectOne(new QueryWrapper<PartInformationEntity>().eq("site", data.getSite()).eq("part_no", data.getComponentPart()));
+        if (partInformationEntity == null) {
+            throw new RuntimeException("子物料不存在!");
+        }
+        data.setPrintUnit(partInformationEntity.getUmId());
         bomManagementMapper.updateBomComponent(data);
         if (dataUrl) {
             // 判断替代是否为正式