Fix exit button which was going away