diff --git a/frontends/tuiapp_v2.py b/frontends/tuiapp_v2.py
index aca93bd1f..abf9d236e 100644
--- a/frontends/tuiapp_v2.py
+++ b/frontends/tuiapp_v2.py
@@ -4767,13 +4767,12 @@ def _on_stream(self, agent_id, task_id, text, done):
                         m.done = True
                         found = m
                         break
-                if found and agent_id == self.current_id:
-                    if found._segment_widgets:
-                        try: self._stream_update_assistant(found)
-                        except Exception: self._refresh_messages()
-                    else:
-                        self._refresh_messages()
-                    if refresh_chrome:
+                    if found and agent_id == self.current_id:
+                        if found._segment_widgets:
+                            try: self._stream_update_assistant(found)
+                            except Exception: self._refresh_messages()
+                        else:
+                            self._refresh_messages()
                         self._refresh_sidebar()
                         self._refresh_topbar()
                     self._ensure_spinner()