Quick fix for badges in buttons