UI bug fix.