diff --git a/frontend/src/components/visual-editor/v2/Diagrams.tsx b/frontend/src/components/visual-editor/v2/Diagrams.tsx
index 9172eac..6903189 100644
--- a/frontend/src/components/visual-editor/v2/Diagrams.tsx
+++ b/frontend/src/components/visual-editor/v2/Diagrams.tsx
@@ -199,6 +199,8 @@ const Diagrams = () => {
       }) => {
         const link = model.getLink(entity.getOptions().id as string);
 
+        if (!link) return;
+
         if (
           !port.getOptions().in ||
           [BlockPorts.nextBlocksOutPort, BlockPorts.attachmentOutPort].includes(