Fix: streaming to tui

1 files changed, 1 insertions, 1 deletions

diff --git a/tui.go b/tui.go
index 9f85a15..7f897e4 100644
--- a/tui.go
+++ b/tui.go
@@ -230,7 +230,6 @@ func initTUI() {
 	tview.Styles = colorschemes["default"]
 	app = tview.NewApplication()
 	pages = tview.NewPages()
-	outputHandler = &TUIOutputHandler{tv: textView}
 	shellInput = tview.NewInputField().
 		SetLabel(fmt.Sprintf("[%s]$ ", cfg.FilePickerDir)). // dynamic prompt
 		SetFieldWidth(0).
@@ -349,6 +348,7 @@ func initTUI() {
 			// calling it explicitly makes text streaming to look more smooth
 			app.Draw()
 		})
+	outputHandler = &TUIOutputHandler{tv: textView}
 	notificationWidget = tview.NewTextView().
 		SetTextAlign(tview.AlignCenter).
 		SetDynamicColors(true).