properly get sensor status from StateMachine (#247)

* properly get sensor status from StateMachine * fixes #246 * forgot the property
FutureTense · Oct 29, 2021 · c5527e9 · c5527e9
1 parent 941b20c
commit c5527e9
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/custom_components/keymaster/system_health.py b/custom_components/keymaster/system_health.py
@@ -16,10 +16,9 @@ def async_register(
 async def system_health_info(hass):
     """Get info for the info page."""
     client = hass.data[DOMAIN]
+    network_sensor = f"binary_sensor.{client['network_sensor']}"
 
     return {
         "zwave_integration": client["zwave_integration"],
-        "network_status": hass.states[
-            f"binary_sensor.{client['network_sensor']}"
-        ].state,
+        "network_status": hass.states.get(network_sensor).state,
     }