diff --git a/packet-service/src/connection_service.rs b/packet-service/src/connection_service.rs
index cafa5f9..e5c2704 100644
--- a/packet-service/src/connection_service.rs
+++ b/packet-service/src/connection_service.rs
@@ -24,6 +24,10 @@ impl ConnectionService {
         self.connections.get(connection_id).map(|entry| entry.clone())
     }
 
+    pub fn get_connection_mut(&self, connection_id: &str) -> Option<dashmap::mapref::one::RefMut<'_, String, ConnectionState>> {
+        self.connections.get_mut(connection_id)
+    }
+
     pub fn remove_connection(&self, connection_id: &str) {
         self.connections.remove(connection_id);
     }