Fix UI after logout