Minor display fix