Fix styling bug