fixing some display issues