diff --git a/app/client.py b/app/client.py
index 265dcea..444f78e 100644
--- a/app/client.py
+++ b/app/client.py
@@ -116,7 +116,7 @@ async def join_server(websocket, system_id, guild_id, channel_id):
 
 # Leave server
 @command
-async def leave_server(guild_id):
+async def leave_server(websocket, guild_id):
     # Takes guild ID
     bot_status = await drb_api.get_bot_status()
     print("Bot status:", bot_status)