diff --git a/dolphinscheduler-server/src/main/java/org/apache/dolphinscheduler/server/master/registry/MasterRegistryClient.java b/dolphinscheduler-server/src/main/java/org/apache/dolphinscheduler/server/master/registry/MasterRegistryClient.java
index e670818f19..906952e709 100644
--- a/dolphinscheduler-server/src/main/java/org/apache/dolphinscheduler/server/master/registry/MasterRegistryClient.java
+++ b/dolphinscheduler-server/src/main/java/org/apache/dolphinscheduler/server/master/registry/MasterRegistryClient.java
@@ -483,6 +483,8 @@ public class MasterRegistryClient {
                 .buildTaskInstanceRelatedInfo(taskInstance)
                 .buildProcessInstanceRelatedInfo(processInstance)
                 .create();
+        taskExecutionContext.setProcessDefineCode(processInstance.getProcessDefinitionCode());
+        taskExecutionContext.setProcessDefineVersion(processInstance.getProcessDefinitionVersion());
 
         if (masterConfig.getMasterKillYarnJobWhenHandleFailOver()
             && !(taskInstance.isSubProcess()