ui bug fix