Change badge style from flat to flat-square