fix some display bugs