diff --git a/bot/xunfei/xunfei_spark_bot.py b/bot/xunfei/xunfei_spark_bot.py
index b5113d4..ddda15e 100644
--- a/bot/xunfei/xunfei_spark_bot.py
+++ b/bot/xunfei/xunfei_spark_bot.py
@@ -178,12 +178,11 @@ class ReplyItem:
 
 # 收到websocket错误的处理
 def on_error(ws, error):
-
     logger.error("[XunFei] error:", error)
 
 
 # 收到websocket关闭的处理
-def on_close(ws):
+def on_close(ws, one, two):
     data_queue = queue_map.get(ws.session_id)
     data_queue.put("END")