diff --git a/menu_traversal_test.go b/menu_traversal_test.go
index ef5f1ab..6cd41de 100644
--- a/menu_traversal_test.go
+++ b/menu_traversal_test.go
@@ -324,27 +324,19 @@ func TestMyAccount_Savefirstname(t *testing.T) {
 		for _, group := range groups {
 			for index, step := range group.Steps {
 				t.Logf("step %v with input %v", index, step.Input)
-
-				cont, err := en.Exec(ctx, []byte("3"))
-
+				_, err := en.Exec(ctx, []byte(step.Input))
 				if err != nil {
 					t.Errorf("Test case '%s' failed at input '%s': %v", group.Name, step.Input, err)
 					return
 				}
-
-				if !cont {
-					break
-				}
 				w := bytes.NewBuffer(nil)
 				if _, err := en.Flush(ctx, w); err != nil {
 					t.Errorf("Test case '%s' failed during Flush: %v", group.Name, err)
 				}
-
 				b := w.Bytes()
 				if !bytes.Equal(b, []byte(step.ExpectedContent)) {
 					t.Fatalf("expected:\n\t%s\ngot:\n\t%s\n", step.ExpectedContent, b)
 				}
-				en.Finish()
 
 			}
 		}