Slight styling regression fix