fixed a little ui problem