ZHA Hibro plug doesn't keep plug state

Just got CC2531 installed with a Hibro plug from Amazon. The plug seems to work at a good distance from my HA server but i’ve noticed HA doesn’t maintain the on or off state of the plug. The standard switch in my GUI will stay on for a second then switches back to the off (unavailable) position. Is this a shortcoming of the specific plug or is this an issue with my HA setup?

Using ZHA integration on:
HassOS 5.1 (SSD setup)
Home Assistant 0.114.3
Rpi 4 4gb