From 278a3c552f02b7e1d167bab1b33c16a6f4b0c446 Mon Sep 17 00:00:00 2001
From: parker <parker.g.britt@gmail.com>
Date: Mon, 23 Jun 2025 14:41:36 +0100
Subject: [PATCH] fix: additional memory guard

---
 src/gui/network/Network.cpp | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/gui/network/Network.cpp b/src/gui/network/Network.cpp
index 2809b12..b241128 100644
--- a/src/gui/network/Network.cpp
+++ b/src/gui/network/Network.cpp
@@ -80,6 +80,7 @@ void Network::deleteEdge(QGraphicsItem* edge)
         prevHoverItem_=nullptr;
     }
     scene_->removeItem(edge);
+    scene_->update();
     delete edge;
 }
 
@@ -224,7 +225,7 @@ void Network::keyPressEvent(QKeyEvent *event)
 
 void Network::highlightEdge(QGraphicsItem* edge, bool state)
 {
-    if(!edge) return;
+    if(!edge || !isType<NodeEdgeGraphic>(edge)) return;
     if(state)
     {
         static_cast<NodeEdgeGraphic*>(edge)->setColor(QColor("red"));