diff --git a/homeassistant/components/hassio/__init__.py b/homeassistant/components/hassio/__init__.py
index 8e8d077c838..5355c5bb16c 100644
--- a/homeassistant/components/hassio/__init__.py
+++ b/homeassistant/components/hassio/__init__.py
@@ -39,10 +39,10 @@ from homeassistant.loader import bind_hass
 from homeassistant.util.async_ import create_eager_task
 from homeassistant.util.dt import now
 
-# config_flow, and entity platforms are imported to ensure
+# config_flow, diagnostics, and entity platforms are imported to ensure
 # other dependencies that wait for hassio are not waiting
 # for hassio to import its platforms
-from . import binary_sensor, config_flow, sensor, update  # noqa: F401
+from . import binary_sensor, config_flow, diagnostics, sensor, update  # noqa: F401
 from .addon_manager import AddonError, AddonInfo, AddonManager, AddonState  # noqa: F401
 from .addon_panel import async_setup_addon_panel
 from .auth import async_setup_auth_view
diff --git a/homeassistant/components/hassio/diagnostics.py b/homeassistant/components/hassio/diagnostics.py
index 41b3fc1c5c3..e4f98e87fc5 100644
--- a/homeassistant/components/hassio/diagnostics.py
+++ b/homeassistant/components/hassio/diagnostics.py
@@ -9,7 +9,8 @@ from homeassistant.config_entries import ConfigEntry
 from homeassistant.core import HomeAssistant
 from homeassistant.helpers import device_registry as dr, entity_registry as er
 
-from . import ADDONS_COORDINATOR, HassioDataUpdateCoordinator
+from .const import ADDONS_COORDINATOR
+from .data import HassioDataUpdateCoordinator
 
 
 async def async_get_config_entry_diagnostics(