[UI] Fix bugs