fixes some UI