fixed the UI