Minor UI edit