change badge arguments type