Fix badge code