add some ui fix