ui bug fixes