diff --git a/src/services/websocket/index.ts b/src/services/websocket/index.ts
index e722e1de..135ea37c 100644
--- a/src/services/websocket/index.ts
+++ b/src/services/websocket/index.ts
@@ -17,7 +17,7 @@ export class WebsocketService {
         this.connection.onmessage = (message: any) => {
             const messageHandler = this.messageHandlers.get(message.type);
             if (messageHandler) {
-                messageHandler(message);
+                messageHandler(message.data);
             } else {
                 console.error(`Unknown websocket message: ${message.type}`);
             }