diff --git a/dolphinscheduler-ui-next/src/views/projects/workflow/components/dag/use-cell-update.ts b/dolphinscheduler-ui-next/src/views/projects/workflow/components/dag/use-cell-update.ts
index 8069ac924e..1cfc1e2da5 100644
--- a/dolphinscheduler-ui-next/src/views/projects/workflow/components/dag/use-cell-update.ts
+++ b/dolphinscheduler-ui-next/src/views/projects/workflow/components/dag/use-cell-update.ts
@@ -86,7 +86,9 @@ export function useCellUpdate(options: Options) {
     const edges = getNodeEdge(id)
     if (edges?.length) {
       edges.forEach((edge) => {
-        graph.value?.removeEdge(edge)
+        if (edge.getTargetNode()?.id === id) {
+          graph.value?.removeEdge(edge)
+        }
       })
     }
     preTaskCode.forEach((task) => {