diff --git a/src/gateway/wasm/heartbeat.rs b/src/gateway/wasm/heartbeat.rs
index b7346fa..86d9573 100644
--- a/src/gateway/wasm/heartbeat.rs
+++ b/src/gateway/wasm/heartbeat.rs
@@ -1,7 +1,9 @@
 use tokio::task::JoinHandle;
+use ws_stream_wasm::*;
 
 use super::*;
 
+/// Handles sending heartbeats to the gateway in another thread
 #[allow(dead_code)] // FIXME: Remove this, once used
 #[derive(Debug)]
 pub struct WasmHeartbeatHandler {