fixed some ui