[Bug] frequent disconnects

[muddro1]

Using master, getting the following two issues:

Logger: custom_components.truenas.coordinator
Source: helpers/update_coordinator.py:347
integration: TrueNAS (documentation, issues)
First occurred: 4:14:55 PM (8 occurrences)
Last logged: 5:23:44 PM

Error fetching TrueNAS data: TrueNas Disconnected

And

Logger: custom_components.truenas.coordinator
Source: helpers/update_coordinator.py:315
integration: TrueNAS (documentation, issues)
First occurred: 4:16:08 PM (13 occurrences)
Last logged: 5:11:25 PM

Unexpected error fetching TrueNAS data: 'root_dataset'
Traceback (most recent call last):
  File "/usr/src/homeassistant/homeassistant/helpers/update_coordinator.py", line 315, in _async_refresh
    self.data = await self._async_update_data()
                ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/config/custom_components/truenas/coordinator.py", line 104, in _async_update_data
    await self.hass.async_add_executor_job(self.get_pool)
  File "/usr/local/lib/python3.12/concurrent/futures/thread.py", line 58, in run
    result = self.fn(*self.args, **self.kwargs)
             ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/config/custom_components/truenas/coordinator.py", line 778, in get_pool
    self.ds["pool"][uid].pop("root_dataset")
KeyError: 'root_dataset'

Not sure if they are related.