diff --git a/files/routes/chat.py b/files/routes/chat.py
index bf243bf65..b5ec9d977 100644
--- a/files/routes/chat.py
+++ b/files/routes/chat.py
@@ -237,9 +237,9 @@ def disconnect(v):
 	if not request.referrer: return ''
 	room = request.referrer
 
-	if request.referrer == f'{SITE_FULL}/notifications/messages':
+	if room == f'{SITE_FULL}/notifications/messages':
 		leave_room(v.id)
-		online["messages"].remove(v.id)
+		online["messages"].discard(v.id)
 		return ''
 
 	if online.get(room):