diff --git a/pkg/tui/views/chat/update.go b/pkg/tui/views/chat/update.go
index 871f1de..66878c5 100644
--- a/pkg/tui/views/chat/update.go
+++ b/pkg/tui/views/chat/update.go
@@ -95,6 +95,7 @@ func (m *Model) Update(msg tea.Msg) (shared.ViewModel, tea.Cmd) {
 			if toEdit.Content != contents {
 				toEdit.Content = contents
 				m.setMessage(m.selectedMessage, toEdit)
+				m.updateContent()
 				if m.persistence && toEdit.ID > 0 {
 					// create clone of message with its new contents
 					cmds = append(cmds, m.cloneMessage(toEdit, true))