fix minor UI