fixing display issue