UI bug fixup