From e052bd065c8f92a886c40ad6f11b05dc73932d93 Mon Sep 17 00:00:00 2001
From: Bojan Mojsilovic <bmojsilovic@itekako.com>
Date: Mon, 11 Sep 2023 16:14:27 +0200
Subject: [PATCH] Fix API change handling

---
 src/contexts/MessagesContext.tsx | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/src/contexts/MessagesContext.tsx b/src/contexts/MessagesContext.tsx
index 5a1b3f3..b269902 100644
--- a/src/contexts/MessagesContext.tsx
+++ b/src/contexts/MessagesContext.tsx
@@ -289,7 +289,13 @@ export const MessagesProvider = (props: { children: ContextChildren }) => {
 
     updateStore('messages', (conv) => [ ...conv, ...newMessages ]);
     resetMessageCount(store.selectedSender, subidResetMsgCount);
-    updateStore('messageCountPerSender', store.selectedSender, 'cnt', 0)
+
+    if (store.messageCountPerSender[store.selectedSender]) {
+      updateStore('messageCountPerSender', store.selectedSender, 'cnt', 0);
+    }
+    else {
+      updateStore('messageCountPerSender', store.selectedSender, () => ({ cnt: 0 }));
+    }
 
     parseForMentions(newMessages);
     then(newMessages);