[Bugs] Fix UI referenced