fix display bugs