diff --git a/websocket.go b/websocket.go
index fa5cfd3..d64645d 100644
--- a/websocket.go
+++ b/websocket.go
@@ -435,3 +435,7 @@ func (wc *WebSocketClient) Stop() {
 	_ = wc.conn.Close()
 	wc.wg.Wait()
 }
+
+func (wc *WebSocketClient) Done() <-chan struct{} {
+	return wc.done
+}