diff --git a/custom_components/alexa_media/sensor.py b/custom_components/alexa_media/sensor.py index e17a3a54..89674f12 100644 --- a/custom_components/alexa_media/sensor.py +++ b/custom_components/alexa_media/sensor.py @@ -248,7 +248,11 @@ async def async_update(self): pass account_dict = (self.hass.data[DATA_ALEXAMEDIA]['accounts'] [self._account]) - self._n_dict = account_dict['notifications'][self._dev_id][self._type] + try: + self._n_dict = (account_dict['notifications'][self._dev_id] + [self._type]) + except KeyError: + self._n_dict = None self._all = (sorted(self._n_dict.items(), key=lambda x: x[1][self._sensor_property]) if self._n_dict else [])