From 054ebf36b5a8f1804c037862990b5f36f5efa634 Mon Sep 17 00:00:00 2001
From: Lew Rossman <LRossman@outlook.com>
Date: Mon, 17 Feb 2025 09:23:35 -0500
Subject: [PATCH] Adjust Trace Node index when node deleted

---
 src/epanet.c | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/src/epanet.c b/src/epanet.c
index 38dee56..cccd107 100644
--- a/src/epanet.c
+++ b/src/epanet.c
@@ -1996,6 +1996,7 @@ int DLLEXPORT EN_deletenode(EN_Project p, int index, int actionCode)
 */
 {
     Network *net = &p->network;
+    Quality  *qual = &p->quality;
 
     int i, nodeType, tankindex;
     Snode *node;
@@ -2041,6 +2042,9 @@ int DLLEXPORT EN_deletenode(EN_Project p, int index, int actionCode)
         // ... update node's entry in the hash table
         hashtable_update(net->NodeHashTable, net->Node[i].ID, i);
     }
+    
+    // Adjust index of water quality trace node
+    if (qual->TraceNode > index) qual->TraceNode -= 1;
 
     // If deleted node is a tank, remove it from the Tank array
     if (nodeType != EN_JUNCTION)