pr fix

custom-components · Apr 6, 2022 · 005d5ac · 005d5ac
1 parent b408075
commit 005d5ac
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/custom_components/zaptec/switch.py b/custom_components/zaptec/switch.py
@@ -17,7 +17,7 @@ async def async_setup_platform(
     hass: HomeAssistantType, config: ConfigType, async_add_entities, discovery_info=None
 ) -> bool:  # pylint: disable=unused-argument
     """Setup switch platform."""
-    acc = hass.data[DOMAIN]["api"]
+    acc = hass.data.get(DOMAIN, {}).get("api")
     if acc is None:
         _LOGGER.debug("Didn't setup switch the api wasnt ready")
         return False