fixed some more display issues