Replace badge with button