diff --git a/static/ui.js b/static/ui.js
index 02c1544c..bac13bbc 100644
--- a/static/ui.js
+++ b/static/ui.js
@@ -11430,6 +11430,7 @@ function renderMessages(options){
       }else{
         currentAssistantTurn=_createAssistantTurn(tsTitle, isTpsDisplayEnabled()?_formatTurnTps(m._turnTps):'');
       }
+      currentAssistantTurn.dataset.msgIdx=rawIdx;
       inner.appendChild(currentAssistantTurn);
     }
     const seg=document.createElement('div');