Change badge to style = flat