diff --git a/src/main.js b/src/main.js
index 59317d4..46496c4 100644
--- a/src/main.js
+++ b/src/main.js
@@ -151,7 +151,7 @@ class MiniGraphCard extends LitElement {
     if (UPDATE_PROPS.some(prop => changedProps.has(prop))) {
       this.color = this.computeColor(
         this.tooltip.value !== undefined
-          ? this.tooltip.value : this.entity[0] && this.entity[0].state,
+          ? this.tooltip.value : this.getEntityState(0),
         this.tooltip.entity || 0,
       );
       return true;