Fix badge helper