diff --git a/src/components/entity/state-badge.html b/src/components/entity/state-badge.html
index 082fbb9bcf..acf7cdd55d 100644
--- a/src/components/entity/state-badge.html
+++ b/src/components/entity/state-badge.html
@@ -14,7 +14,7 @@
     border-radius: 50%;
   }
 
-  div {
+  .badge {
     height: 40px;
     text-align: center;
   }
@@ -37,7 +37,7 @@
   </style>
 
   <template>
-    <div class='layout horizontal center'>
+    <div class='layout horizontal center badge'>
       <domain-icon id='icon'
         domain='[[stateObj.domain]]' data-domain$='[[stateObj.domain]]'
         state='[[stateObj.state]]' data-state$='[[stateObj.state]]'>