diff --git a/src/snek/static/message-list.js b/src/snek/static/message-list.js
index ea20826..e4984bb 100644
--- a/src/snek/static/message-list.js
+++ b/src/snek/static/message-list.js
@@ -280,11 +280,15 @@ class MessageList extends HTMLElement {
   upsertMessage(data) {
     let message = this.messageMap.get(data.uid);
     if (message && (data.is_final || !data.message)) {
-      message.parentElement?.removeChild(message);
+      //message.parentElement?.removeChild(message);
       // TO force insert 
-      message = null;
+      //message = null;
       
     }
+    if(message && !data.message){
+      message.parentElement?.removeChild(message);
+      message = null;
+    }
     if (!data.message) return;
 
     const wrapper = document.createElement("div");