Minor display bug correction