diff --git a/dinky-admin/src/main/java/org/dinky/controller/SchedulerController.java b/dinky-admin/src/main/java/org/dinky/controller/SchedulerController.java
index 212974d0c3..87387340a1 100644
--- a/dinky-admin/src/main/java/org/dinky/controller/SchedulerController.java
+++ b/dinky-admin/src/main/java/org/dinky/controller/SchedulerController.java
@@ -207,6 +207,9 @@ public Result<String> createTaskDefinition(
             return Result.failed(Status.DS_WORK_FLOW_DEFINITION_TASK_NAME_EXIST, processName, taskName);
         }
 
+        Long taskCode = taskClient.genTaskCode(projectCode);
+        taskRequest.setCode(taskCode);
+
         String taskDefinitionJsonObj = JSONUtil.toJsonStr(taskRequest);
         taskClient.createTaskDefinition(projectCode, process.getCode(), upstreamCodes, taskDefinitionJsonObj);