Fix styling regression