diff --git a/src/state.rs b/src/state.rs
index a391401..99e5369 100644
--- a/src/state.rs
+++ b/src/state.rs
@@ -146,6 +146,7 @@ mod tests {
             state.blocked.insert(ip1);
             assert!(state.blocked.contains(&ip1));
             assert!(!state.blocked.contains(&ip2));
+            drop(state);
         }
 
         // Test active connections
@@ -159,6 +160,7 @@ mod tests {
             assert_eq!(state.active_connections.len(), 1);
             assert!(!state.active_connections.contains(&ip1));
             assert!(state.active_connections.contains(&ip2));
+            drop(state);
         }
     }
 }