fixed the ui