update: improve logging

lareii · Dec 6, 2023 · 0c166b0 · 0c166b0
1 parent a311c9c
commit 0c166b0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/main.go b/main.go
@@ -40,9 +40,9 @@ func init() {
 		var ok string
 		fmt.Scan(&ok)
 
-		if ok == "y" {
+		if ok[0] == 'y' {
 			break
-		} else if ok == "n" {
+		} else if ok[0] == 'n' {
 			os.Exit(0)
 		} else {
 			println("nuh uh")