From 0d4060bb97e962493aa15c68c0fd61f2685c7865 Mon Sep 17 00:00:00 2001
From: Paulus Schoutsen <paulus@paulusschoutsen.nl>
Date: Tue, 31 Jan 2017 07:33:33 -0800
Subject: [PATCH] Fix passing in non-existing domains

---
 src/components/entity/ha-state-label-badge.html | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/src/components/entity/ha-state-label-badge.html b/src/components/entity/ha-state-label-badge.html
index 23ddd1052b..38e8e939be 100644
--- a/src/components/entity/ha-state-label-badge.html
+++ b/src/components/entity/ha-state-label-badge.html
@@ -88,7 +88,8 @@ Polymer({
     if (state.state === 'unavailable') {
       return null;
     }
-    switch (window.hassUtil.computeDomain(state)) {
+    var domain = window.hassUtil.computeDomain(state);
+    switch (domain) {
       case 'alarm_control_panel':
         if (state.state === 'pending') {
           return 'mdi:clock-fast';
@@ -100,14 +101,14 @@ Polymer({
           return 'mdi:alert-circle';
         }
         // state == 'disarmed'
-        return window.hassUtil.domainIcon(state.domain, state.state);
+        return window.hassUtil.domainIcon(domain, state.state);
       case 'binary_sensor':
       case 'device_tracker':
       case 'updater':
         return window.hassUtil.stateIcon(state);
       case 'sun':
         return state.state === 'above_horizon' ?
-        window.hassUtil.domainIcon(state.domain) : 'mdi:brightness-3';
+        window.hassUtil.domainIcon(domain) : 'mdi:brightness-3';
       default:
         return null;
     }