fix minor UI bug