diff --git a/src/frontend/src/modals/IOModal/components/chatView/chat-view.tsx b/src/frontend/src/modals/IOModal/components/chatView/chat-view.tsx
index 49d9be259..a278d7587 100644
--- a/src/frontend/src/modals/IOModal/components/chatView/chat-view.tsx
+++ b/src/frontend/src/modals/IOModal/components/chatView/chat-view.tsx
@@ -23,7 +23,8 @@ const MemoizedChatMessage = memo(ChatMessage, (prevProps, nextProps) => {
     prevProps.chat.id === nextProps.chat.id &&
     prevProps.chat.session === nextProps.chat.session &&
     prevProps.chat.content_blocks === nextProps.chat.content_blocks &&
-    prevProps.chat.properties === nextProps.chat.properties
+    prevProps.chat.properties === nextProps.chat.properties &&
+    prevProps.lastMessage === nextProps.lastMessage
   );
 });