Fix a little ui problem