jo

1 files changed, 3 insertions, 4 deletions

diff --git a/internal/prompt/prompt.go b/internal/prompt/prompt.go
index bf5997d..6eecf60 100644
--- a/internal/prompt/prompt.go
+++ b/internal/prompt/prompt.go
@@ -46,8 +46,7 @@ func whatNow(question string) error {
 			continue
 		}
 
-		input = strings.TrimSpace(input)
-		switch strings.ToLower(input) {
+		switch strings.ToLower(strings.TrimSpace(input)) {
 		case "y", "yes":
 			return nil
 		case "n", "no":
@@ -61,8 +60,8 @@ func whatNow(question string) error {
 }
 
 func EditFile(filePath string) error {
-	editor := os.Getenv("EDITOR")
-	if editor == "" {
+	editor, ok := os.LookupEnv("EDITOR")
+	if !ok {
 		return errors.New("EDITOR environment variable is not set")
 	}