Ui bug fixes