diff --git a/src/util/hass-util.html b/src/util/hass-util.html
index 02ab9cc1be..5216fccf17 100644
--- a/src/util/hass-util.html
+++ b/src/util/hass-util.html
@@ -323,7 +323,7 @@ window.hassUtil.domainIcon = function (domain, state) {
 
 window.hassUtil.binarySensorIcon = function (state) {
   var activated = state.state && state.state === 'off';
-  switch (state.attributes.sensor_class) {
+  switch (state.attributes.device_class) {
     case 'connectivity':
       return activated ? 'mdi:server-network-off' : 'mdi:server-network';
     case 'light':