diff --git a/src/frontend/src/util/reactflowUtils.ts b/src/frontend/src/util/reactflowUtils.ts
index d09ac3455..77e8e0cdd 100644
--- a/src/frontend/src/util/reactflowUtils.ts
+++ b/src/frontend/src/util/reactflowUtils.ts
@@ -45,8 +45,8 @@ export function cleanEdges({
         }
       }
     }
-    if (changed) {
-      updateEdge(newEdges);
-    }
   });
+  if (changed) {
+    updateEdge(newEdges);
+  }
 }