Fixed another display bug