[Bugs] Fix UI