fix some ui problems