Correct small display bug