diff --git a/ktg-mes/src/main/java/com/ktg/mes/wm/controller/WmsInTaskController.java b/ktg-mes/src/main/java/com/ktg/mes/wm/controller/WmsInTaskController.java
index d56f45f..657c363 100644
--- a/ktg-mes/src/main/java/com/ktg/mes/wm/controller/WmsInTaskController.java
+++ b/ktg-mes/src/main/java/com/ktg/mes/wm/controller/WmsInTaskController.java
@@ -220,7 +220,7 @@ public class WmsInTaskController extends BaseController {
                     baseKnife.setCreateTime(DateUtils.getNowDate());
                     baseKnifeService.updateBaseKnife(baseKnife);
                 });
-            } else if (Objects.equals(wmsInTask.getPlanTypeId(), "9")) {
+            } else if (Objects.equals(wmsInTask.getPlanTypeId(), "9") || Objects.equals(wmsInTask.getPlanTypeCode(), "ZDHK")) {
                 // 入库类型为整刀回库
                 wmsInPlanDetailEntityList.forEach(planDetailEntity -> {
                     BaseKnife baseKnife = baseKnifeService.selectBaseKnifeByBaseKnifeId(planDetailEntity.getKnifeId());
@@ -283,7 +283,7 @@ public class WmsInTaskController extends BaseController {
                         }
                     }
                 });
-            } else if (Objects.equals(wmsInTask.getPlanTypeId(), "15")) {
+            } else if (Objects.equals(wmsInTask.getPlanTypeId(), "15") || Objects.equals(wmsInTask.getPlanTypeCode(), "CDHK")) {
                 // 拆刀回库
                 wmsInPlanDetailEntityList.forEach(planDetailEntity -> {
                     BaseKnife baseKnife = baseKnifeService.selectBaseKnifeByBaseKnifeId(planDetailEntity.getKnifeId());
@@ -300,7 +300,7 @@ public class WmsInTaskController extends BaseController {
                         baseKnifeService.updateBaseKnife(bladeOpt.get());
                     }
                 });
-            } else if (Objects.equals(wmsInTask.getPlanTypeId(), "7")) {
+            } else if (Objects.equals(wmsInTask.getPlanTypeId(), "7") || Objects.equals(wmsInTask.getPlanTypeCode(), "DBRK")) {
                 // 入库类型为刀柄入库
                 wmsInPlanDetailEntityList.forEach(planDetailEntity -> {
                     // 生成台账