diff --git a/src/websocket.js b/src/websocket.js
index ed4b420..d7d2afc 100644
--- a/src/websocket.js
+++ b/src/websocket.js
@@ -40,7 +40,7 @@ function createWebSocket(token, onUpdate, onMessage) {
     console.log("WebSocket 已关闭", result);
     onUpdate(false);
     stopHeartbeat();
-    reconnect(onMessage);
+    reconnect(onUpdate, onMessage);
   });
 
   // 启动心跳
@@ -50,7 +50,7 @@ function createWebSocket(token, onUpdate, onMessage) {
 /**
  * 重连机制
  */
-function reconnect(onMessage) {
+function reconnect(onUpdate, onMessage) {
   if (reconnectCount >= RECONNECT_CONFIG.MAX_COUNT) return;
 
   reconnectTimer && clearTimeout(reconnectTimer);
@@ -64,7 +64,7 @@ function reconnect(onMessage) {
   );
   reconnectTimer = setTimeout(() => {
     console.log("reconnecting...");
-    createWebSocket(token, onMessage);
+    createWebSocket(token, onUpdate, onMessage);
     reconnectCount++;
   }, delay);
 }