Fix bug UI

Fix bug UI