diff --git a/test4/main.go b/test4/main.go
index c09abd4..b95d8fe 100644
--- a/test4/main.go
+++ b/test4/main.go
@@ -22,7 +22,10 @@ func buttonClick(i int, s string) {
 		return
 	} else if (gui.Data.State == "account1") {
 		gui.ShowAccountTab()
-		gui.Data.State = "account2"
+		gui.Data.State = "main"
+	} else if (gui.Data.State == "main") {
+		gui.ShowMainTab()
+		gui.Data.State = "done"
 	}
 }