B158 Switch Swap

Morning all, just wondering if anyone knows if its possible to get a straight up replacement switch that’ll just drop into the B158? Looking for a reverse clicky rather than forward though.

Just fitted the A17DD driver to bump its output which works a treat. Only problem is the driver is designed around using half presses to change modes etc which doesn’t work with the standard B158 switch making the UI rather clumsy.