Fixing Xiaomi Piston 2 headset support

Might fix some other multi-button headsets as well