Fix UI bug.