Fix startup with 0.118dev (#1623)

hacs · Nov 4, 2020 · 4d5eecb · 4d5eecb
1 parent 967cecc
commit 4d5eecb
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/custom_components/hacs/operational/setup.py b/custom_components/hacs/operational/setup.py
@@ -2,7 +2,6 @@
 from custom_components.hacs.enums import HacsStage
 from aiogithubapi import AIOGitHubAPIException, GitHub
 from homeassistant import config_entries
-from homeassistant.components.lovelace import system_health_info
 from homeassistant.const import EVENT_HOMEASSISTANT_STARTED
 from homeassistant.const import __version__ as HAVERSION
 from homeassistant.exceptions import ConfigEntryNotReady
@@ -33,6 +32,11 @@
 )
 from custom_components.hacs.share import get_hacs
 
+try:
+    from homeassistant.components.lovelace import system_health_info
+except ImportError:
+    from homeassistant.components.lovelace.system_health import system_health_info
+
 
 async def _async_common_setup(hass):
     """Common setup stages."""