Add command to log out

1 files changed, 2 insertions, 0 deletions

diff --git a/ui/ui.go b/ui/ui.go
index 2f45f27..5e06d16 100644
--- a/ui/ui.go
+++ b/ui/ui.go
@@ -78,10 +78,12 @@ func (ui *GomuksUI) Render() {
 
 func (ui *GomuksUI) OnLogin() {
 	ui.SetView(ViewMain)
+	ui.app.SetFocus(ui.mainView)
 }
 
 func (ui *GomuksUI) OnLogout() {
 	ui.SetView(ViewLogin)
+	ui.app.SetFocus(ui.loginView)
 }
 
 func (ui *GomuksUI) SetView(name View) {