Minor UI bug fix