bug fixes UI