Ui bug fix