diff --git a/dolphinscheduler-ui/src/views/projects/task/components/node/fields/use-environment-name.ts b/dolphinscheduler-ui/src/views/projects/task/components/node/fields/use-environment-name.ts
index 772079d418..dd51b8d1aa 100644
--- a/dolphinscheduler-ui/src/views/projects/task/components/node/fields/use-environment-name.ts
+++ b/dolphinscheduler-ui/src/views/projects/task/components/node/fields/use-environment-name.ts
@@ -29,7 +29,7 @@ export function useEnvironmentName(
   let environmentList = [] as IEnvironmentNameOption[]
   const options = ref([] as IEnvironmentNameOption[])
   const loading = ref(false)
-  const value = ref()
+  let mounted = false
 
   const getEnvironmentList = async () => {
     if (loading.value) return
@@ -46,6 +46,11 @@ export function useEnvironmentName(
       filterByWorkerGroup(option)
     )
     loading.value = false
+    if (options.value.length === 0) {
+      model.environmentCode = null
+    } else {
+      isCreate && (model.environmentCode = options.value[0].value)
+    }
   }
 
   const filterByWorkerGroup = (option: IEnvironmentNameOption) => {
@@ -54,28 +59,21 @@ export function useEnvironmentName(
     return option.workerGroups.indexOf(model.workerGroup) !== -1
   }
 
-  watch(
-    () => options.value.length,
-    () => {
-      if (isCreate && options.value.length === 1 && !value.value) {
-        model.environmentCode = options.value[0].value
-      }
-      if (options.value.length === 0) model.environmentCode = null
-    }
-  )
-
   watch(
     () => model.workerGroup,
     () => {
-      if (!model.workerGroup) return
+      if (!model.workerGroup || !mounted) return
       options.value = environmentList.filter((option: IEnvironmentNameOption) =>
         filterByWorkerGroup(option)
       )
+      model.environmentCode =
+        options.value.length === 0 ? null : options.value[0].value
     }
   )
 
-  onMounted(() => {
-    getEnvironmentList()
+  onMounted(async () => {
+    await getEnvironmentList()
+    mounted = true
   })
 
   return {
diff --git a/dolphinscheduler-ui/src/views/projects/workflow/components/dag/use-task-edit.ts b/dolphinscheduler-ui/src/views/projects/workflow/components/dag/use-task-edit.ts
index 9f305825c0..dfe7b34ee3 100644
--- a/dolphinscheduler-ui/src/views/projects/workflow/components/dag/use-task-edit.ts
+++ b/dolphinscheduler-ui/src/views/projects/workflow/components/dag/use-task-edit.ts
@@ -174,7 +174,8 @@ export function useTaskEdit(options: Options) {
             ...taskDef,
             version: task.version,
             code: task.code,
-            taskType: currTask.value.taskType
+            taskType: currTask.value.taskType,
+            id: task.id
           }
         }
         return task