Fixed a small display bug