fix some ui bug