Fix relogin bugs