diff --git a/main.go b/main.go
index 55db5e4..6bd0532 100644
--- a/main.go
+++ b/main.go
@@ -198,6 +198,8 @@ input:
 			for {
 				input, err = continuePrompt(input)
 				if err != nil {
+					running = true
+					lr.SetPrompt(fmtPrompt(prompt))
 					goto input // continue inside nested loop
 				}
 				if !strings.HasSuffix(input, "\\") {