Fixed minor display bug