Unifi Network does not reconnect after controller restart

I recently noticed, that when the Unifi controller restarts (e.g. after updating), the Unifi Network Home Assistant Integration does not reconnect to the controller. I had two ocurrences in the last days (log4j update from Ubiquiti) where I had to manually restart Home Assistant to get it reconnected.

Maybe the integration could use some kind of watchdog mechanism to monitor the controller connection and reconnect if necessary.

This issue still seems to be present.
I need to reload the integration (HA) once i updated my controller or restarted it.
Not a big issue at all but sometimes you forget this and as many will use the Unifi to track devices, automations won’t trigger.