diff --git a/src/frontend/src/utils/reactflowUtils.ts b/src/frontend/src/utils/reactflowUtils.ts
index 7aa6a6ff1..80d9f11a0 100644
--- a/src/frontend/src/utils/reactflowUtils.ts
+++ b/src/frontend/src/utils/reactflowUtils.ts
@@ -48,6 +48,7 @@ export function checkChatInput(nodes: Node[]) {
 }
 
 export function cleanEdges(nodes: NodeType[], edges: Edge[]) {
+  console.log("cleanEdges", nodes, edges);
   let newEdges = cloneDeep(edges);
   edges.forEach((edge) => {
     // check if the source and target node still exists
@@ -98,6 +99,9 @@ export function cleanEdges(nodes: NodeType[], edges: Edge[]) {
           newEdges = newEdges.filter((e) => e.id !== edge.id);
         }
       }
+      else {
+        newEdges = newEdges.filter((e) => e.id !== edge.id);
+      }
     }
   });
   return newEdges;