Fix display issue