Tried to correct display bug