Fix node remove logic

19f5d7afa87d9642a2588d5de97fddc16ba71154 · ruwaq Marcin Jaworski · 2019-09-05T00:22:39Z (7 wata ñawpaq) · tayta 03b0a161d · kay commitpi sach'ata qhaway

Semántico sayk'uy · 1

src/node.rs

ukhunpi t'ikrakurqan fn remove pub

1 qillqakuna · +1 −0

src/node.rs +1 −0

diff --git a/src/node.rs b/src/node.rs
index 0ab316000..b312503c0 100644
--- a/src/node.rs
+++ b/src/node.rs
@@ -108,6 +108,7 @@ impl Stretch {
         if let Some(new_id) = self.forest.swap_remove(id) {
             let new = self.ids_to_nodes.remove(&new_id).unwrap();
             self.nodes_to_ids.insert(new, id);
+            self.ids_to_nodes.insert(id, new);
         }
     }