diff --git a/src/main/java/neatlogic/module/process/stephandler/utilhandler/AutomaticProcessUtilHandler.java b/src/main/java/neatlogic/module/process/stephandler/utilhandler/AutomaticProcessUtilHandler.java
index 7586a997d7235891197a9837ede3b4f297be6f89..2fd7cd53a6e2fb00c3bb0bd20fac2f7a7d14077f 100644
--- a/src/main/java/neatlogic/module/process/stephandler/utilhandler/AutomaticProcessUtilHandler.java
+++ b/src/main/java/neatlogic/module/process/stephandler/utilhandler/AutomaticProcessUtilHandler.java
@@ -2,6 +2,7 @@ package neatlogic.module.process.stephandler.utilhandler;
 
 import com.alibaba.fastjson.JSONObject;
 import neatlogic.framework.common.constvalue.systemuser.SystemUser;
+import neatlogic.framework.notify.core.INotifyPolicyHandler;
 import neatlogic.framework.process.constvalue.ProcessStepHandlerType;
 import neatlogic.framework.process.constvalue.ProcessTaskOperationType;
 import neatlogic.framework.process.constvalue.ProcessTaskStatus;
@@ -12,6 +13,7 @@ import neatlogic.framework.process.operationauth.core.IOperationType;
 import neatlogic.framework.process.operationauth.core.ProcessAuthManager;
 import neatlogic.framework.process.stephandler.core.ProcessStepInternalHandlerBase;
 import neatlogic.module.process.dao.mapper.processtask.ProcessTaskStepDataMapper;
+import neatlogic.module.process.notify.handler.AutomaticNotifyPolicyHandler;
 import org.springframework.stereotype.Service;
 
 import javax.annotation.Resource;
@@ -86,6 +88,11 @@ public class AutomaticProcessUtilHandler extends ProcessStepInternalHandlerBase
         };
     }
 
+    @Override
+    public Class<? extends INotifyPolicyHandler> getNotifyPolicyHandlerClass() {
+        return AutomaticNotifyPolicyHandler.class;
+    }
+
     @Override
     public String[] getRegulateKeyList() {
         return new String[]{"authorityList", "notifyPolicyConfig", "customButtonList", "customStatusList", "replaceableTextList", "workerPolicyConfig", "formSceneUuid", "formSceneName", "tagList", "automaticConfig", "formTag"};
diff --git a/src/main/java/neatlogic/module/process/stephandler/utilhandler/OmnipotentProcessUtilHandler.java b/src/main/java/neatlogic/module/process/stephandler/utilhandler/OmnipotentProcessUtilHandler.java
index 8d8bed0ab24436d6b7a741eadbc19876b0f702a6..1bc00e544761055349b48743620c803b205dee3e 100644
--- a/src/main/java/neatlogic/module/process/stephandler/utilhandler/OmnipotentProcessUtilHandler.java
+++ b/src/main/java/neatlogic/module/process/stephandler/utilhandler/OmnipotentProcessUtilHandler.java
@@ -1,5 +1,6 @@
 package neatlogic.module.process.stephandler.utilhandler;
 
+import neatlogic.framework.notify.core.INotifyPolicyHandler;
 import neatlogic.framework.process.constvalue.ProcessStepHandlerType;
 import neatlogic.framework.process.constvalue.ProcessTaskOperationType;
 import neatlogic.framework.process.constvalue.ProcessTaskStepOperationType;
@@ -11,6 +12,7 @@ import neatlogic.framework.process.dto.ProcessTaskStepVo;
 import neatlogic.framework.process.operationauth.core.IOperationType;
 import neatlogic.framework.process.stephandler.core.IProcessStepAssistantHandler;
 import neatlogic.framework.process.stephandler.core.ProcessStepInternalHandlerBase;
+import neatlogic.module.process.notify.handler.OmnipotentNotifyPolicyHandler;
 import org.apache.commons.collections4.CollectionUtils;
 import org.springframework.stereotype.Service;
 
@@ -65,6 +67,11 @@ public class OmnipotentProcessUtilHandler extends ProcessStepInternalHandlerBase
         };
     }
 
+    @Override
+    public Class<? extends INotifyPolicyHandler> getNotifyPolicyHandlerClass() {
+        return OmnipotentNotifyPolicyHandler.class;
+    }
+
     @Override
     public String[] getRegulateKeyList() {
         return new String[]{"authorityList", "notifyPolicyConfig", "actionConfig", "customButtonList", "customStatusList", "replaceableTextList", "workerPolicyConfig", "taskConfig", "enableReapproval", "autoComplete", "autoApproval", "formSceneUuid", "formSceneName", "autoStart", "isNeedUploadFile", "isNeedContent", "isRequired", "commentTemplateId", "tagList"};