[UI] Fix graphical bugs