diff --git a/cmd/cmd.go b/cmd/cmd.go
index 5f2afb0d..9561b81c 100644
--- a/cmd/cmd.go
+++ b/cmd/cmd.go
@@ -398,7 +398,7 @@ func generate(cmd *cobra.Command, model, prompt string, wordWrap bool, format st
 		generateContext = []int{}
 	}
 
-	termWidth, _, err := term.GetSize(int(0))
+	termWidth, _, err := term.GetSize(int(os.Stdout.Fd()))
 	if err != nil {
 		wordWrap = false
 	}