fix display issue