Fix styling on flash