fix some visual bugs