diff --git a/notification_sound.py b/notification_sound.py
index 34152a7..a3e9e60 100644
--- a/notification_sound.py
+++ b/notification_sound.py
@@ -18,7 +18,7 @@ def on_message(ws, message):
     playsound(sound)
 def on_open(ws):
     print('Websocket opened')
-def on_close(ws):
+def on_close(ws, *error):
     print('Websocket closed, attempting to reconnect')
     connect_websocket()