Fix graphical display issues