Safer check

thibmaek · Dec 10, 2018 · e1b00f1 · e1b00f1
1 parent 120172c
commit e1b00f1
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/homeassistant/components/sensor/nmbs.py b/homeassistant/components/sensor/nmbs.py
@@ -159,9 +159,10 @@ def unit_of_measurement(self):
     @property
     def icon(self):
         """Return the sensor default icon or an alert icon if any delay."""
-        delay = get_delay_in_minutes(self._attrs['departure']['delay'])
-        if self._attrs is not None and delay > 0:
-            return "mdi:alert-octagon"
+        if self._attrs is not None:
+            delay = get_delay_in_minutes(self._attrs['departure']['delay'])
+            if delay > 0:
+                return "mdi:alert-octagon"
 
         return "mdi:train"