fix minor UI problem