diff --git a/internal/ui/chat/msg.go b/internal/ui/chat/msg.go
index b90dc78..c3e4b9e 100644
--- a/internal/ui/chat/msg.go
+++ b/internal/ui/chat/msg.go
@@ -20,12 +20,12 @@ func (m *Model) openState() tview.Cmd {
 }
 
 func (m *Model) closeState() tview.Cmd {
+	if m.state == nil {
+		return nil
+	}
 	return func() tview.Msg {
-		if m.state != nil {
-			if err := m.state.Close(); err != nil {
-				slog.Error("failed to close the session", "err", err)
-				return nil
-			}
+		if err := m.state.Close(); err != nil {
+			slog.Error("failed to close the session", "err", err)
 		}
 		return nil
 	}