diff --git a/src/client/WebSocketClient.test.ts b/src/client/WebSocketClient.test.ts
index 26639472..c0d20913 100644
--- a/src/client/WebSocketClient.test.ts
+++ b/src/client/WebSocketClient.test.ts
@@ -215,7 +215,9 @@ describe('WebSocketClient', () => {
           ws.disconnect();
         }
       });
-      ws.on(WebSocketEvent.ON_CLOSE, done);
+      ws.on(WebSocketEvent.ON_CLOSE, () => {
+        done();
+      });
       ws.on(WebSocketEvent.ON_MESSAGE_ERROR, (wsError: WebSocketErrorMessage) => done.fail(wsError.message));
       // Send subscription once the WebSocket is ready
       ws.on(WebSocketEvent.ON_OPEN, () => ws.subscribe(channels));