fix little ui bug