diff --git a/website/src/lib/stores/websocket.ts b/website/src/lib/stores/websocket.ts
index eb34795..54d38b2 100644
--- a/website/src/lib/stores/websocket.ts
+++ b/website/src/lib/stores/websocket.ts
@@ -244,6 +244,16 @@ function connect(): void {
         isConnectedStore.set(true);
         clearReconnectTimer();
         subscribeToChannels();
+        
+        USER_DATA.subscribe(user => {
+            if (user?.id && isSocketConnected()) {
+                console.log('Setting user subscription for user:', user.id);
+                socket!.send(JSON.stringify({
+                    type: 'set_user',
+                    userId: String(user.id)
+                }));
+            }
+        })();
     };
 
     socket.onmessage = handleWebSocketMessage;