Sonoff S40 zigbee outlet claims to save state after power outage

Sonoff says it can save the on/off state after a power outage. I have searched the data base on here and found mention that this is true, however no mention of how to do it. Has anyone done it? I can see how it might be done on a wifi device using the ewelink app but I don't see how it might be done using only Hubitat and a zigbee device. Maybe there is a custom driver?

I know nothing about this particular Sonoff device, but...
When devices have the option to save the power off state, you can usually set that option in the Settings of the Driver (under the Devices tab of Hubitat). If the driver you are using does not show that option in the Settings, perhaps you can find another Driver for that device that does.