fix display issues