Mi Wireless Switch WXKG01LM not showing switch entities

Hi.

I manage to pair the Mi Wireless Switch WXKG01LM to zha but is only showing the battery entites, no switch entites is detected. I am using a sonoff off zigbee bridge flashed with tasmota.

Any ideas?