actually fixed display issue