quick UI bug fix