clean up

marc2332 · Jan 9, 2025 · cf115d7 · cf115d7
1 parent e9f8853
commit cf115d7
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 7 deletions.
diff --git a/crates/torin/examples/demo.rs b/crates/torin/examples/demo.rs
@@ -1,7 +1,4 @@
-use std::{
-    collections::HashMap,
-    sync::Arc,
-};
+use std::collections::HashMap;
 
 use torin::prelude::*;
 

diff --git a/crates/torin/src/test_utils.rs b/crates/torin/src/test_utils.rs
@@ -18,14 +18,12 @@ impl TestingDOM {
     }
 
     pub fn remove(&mut self, node_id: usize) {
-        let node = self.mapper.get(&node_id).unwrap().clone();
+        let node = self.mapper.remove(&node_id).unwrap();
 
         if let Some((_, parent_children, _, _)) = node.0.and_then(|p| self.mapper.get_mut(&p)) {
             parent_children.retain(|c| *c != node_id);
         }
 
-        self.mapper.remove(&node_id);
-
         for child in node.1 {
             self.remove(child);
         }