Fixed the problem with ui.