From 4d55029f8e5c9e760c20396cb7a55af5a8c2cd91 Mon Sep 17 00:00:00 2001
From: Marcin Podlodowski <marcin@podlodowski.it>
Date: Fri, 6 Aug 2021 19:04:10 +0200
Subject: [PATCH] Remove ws connection from WebSocketWatchDog on normal close

---
 src/main/java/com/huobi/utils/WebSocketWatchDog.java | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/main/java/com/huobi/utils/WebSocketWatchDog.java b/src/main/java/com/huobi/utils/WebSocketWatchDog.java
index 644875cd..471eb0b2 100644
--- a/src/main/java/com/huobi/utils/WebSocketWatchDog.java
+++ b/src/main/java/com/huobi/utils/WebSocketWatchDog.java
@@ -58,6 +58,6 @@ public static void onConnectionCreated(WebSocketConnection connection) {
   }
 
   public static void onClosedNormally(WebSocketConnection connection) {
-    TIME_HELPER.remove(connection);
+    TIME_HELPER.remove(connection.getConnectionId());
   }
 }