diff --git a/src/lib/helpers/conversation.js b/src/lib/helpers/conversation.js
index fe181ec8..6c17fb5e 100644
--- a/src/lib/helpers/conversation.js
+++ b/src/lib/helpers/conversation.js
@@ -2,6 +2,7 @@ import { conversationUserStateStore } from "./store";
 
 /**
  * @param {string} conversationId
+ * @returns {import('$conversationTypes').ConversationStateModel[]}
  */
 export function buildConversationUserStates(conversationId) {
     const userStates = conversationUserStateStore.get(conversationId);
diff --git a/src/lib/services/realtime-chat-service.js b/src/lib/services/realtime-chat-service.js
index 9554dd19..027ff02b 100644
--- a/src/lib/services/realtime-chat-service.js
+++ b/src/lib/services/realtime-chat-service.js
@@ -45,7 +45,9 @@ export const realtimeChat = {
             socket?.send(JSON.stringify({
                 event: "start",
                 body: {
-                    payload: JSON.stringify(buildConversationUserStates(conversationId))
+                    payload: JSON.stringify({
+                        states: buildConversationUserStates(conversationId)
+                    })
                 }
             }));