ui bug fix.