diff --git a/pkg/tui/tui.go b/pkg/tui/tui.go
index cb72484..dee11b1 100644
--- a/pkg/tui/tui.go
+++ b/pkg/tui/tui.go
@@ -124,8 +124,8 @@ func (m model) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
 		m.input.SetWidth(msg.Width - 1)
 		m.updateContent()
 	case msgConversationLoaded:
-		c := (*models.Conversation)(msg)
-		cmd = m.loadMessages(c)
+		m.conversation = (*models.Conversation)(msg)
+		cmd = m.loadMessages(m.conversation)
 	case msgMessagesLoaded:
 		m.messages = []models.Message(msg)
 		m.updateContent()