diff --git a/web/src/components/Playground.js b/web/src/components/Playground.js
index c080e13..459cfb0 100644
--- a/web/src/components/Playground.js
+++ b/web/src/components/Playground.js
@@ -144,7 +144,7 @@ const Playground = () => {
 
     source.addEventListener("error", (e) => {
       generateMockResponse(e.data)
-      completeMessage();
+      completeMessage('error')
     });
 
     source.addEventListener("readystatechange", (e) => {
@@ -206,12 +206,17 @@ const Playground = () => {
     });
   }, [getSystemMessage]);
 
-  const completeMessage = useCallback(() => {
+  const completeMessage = useCallback((status = 'complete') => {
+    // console.log("Complete Message: ", status)
     setMessage((prevMessage) => {
       const lastMessage = prevMessage[prevMessage.length - 1];
+      // only change the status if the last message is not complete and not error
+      if (lastMessage.status === 'complete' || lastMessage.status === 'error') {
+        return prevMessage;
+      }
       return [
         ...prevMessage.slice(0, -1),
-        { ...lastMessage, status: 'complete' }
+        { ...lastMessage, status: status }
       ];
     });
   }, [])