ZHA + Xiaomi ZNCZ02LM + Socket state after power outage - always on when ac returns

Hi Everyone

I have ZHA setup with HASS OS. I have about 6 of these Zigbee plugs. Everything works just fine. I want to use one of these plugs on my desktop computer.

How do I configure these things to always switch following a ac power outage? Of the 8 or so that I have, 3 always turn back on when the power comes back, but the others do not.

I know its possible because years ago, I recall finding some instructions on some website on how to do it. It involved sending raw commands to the device but I can’t seem to locate it currently.