minor bug fixes for UI