[Bugs] Fix UI bugs