minor UI fixed