fix some small UI issue