From cd2278b0f07f1f224e51aeda4085d35b850eda5b Mon Sep 17 00:00:00 2001
From: Mindfreeze <mindfreeze@abv.bg>
Date: Wed, 18 Jan 2023 16:00:35 +0200
Subject: [PATCH] fix(#60): type fix

---
 src/ha-sankey-chart.ts | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/ha-sankey-chart.ts b/src/ha-sankey-chart.ts
index 5e24530..e48d5ea 100644
--- a/src/ha-sankey-chart.ts
+++ b/src/ha-sankey-chart.ts
@@ -176,7 +176,7 @@ export class SankeyChart extends SubscribeMixin(LitElement) {
     const deviceIds = (collection.prefs?.device_consumption || [])
       .map(d => d.stat_consumption)
       .filter(id => {
-        if (!this.hass.entities[id]) {
+        if (!this.hass.states[id]) {
           console.warn('Ignoring missing entity ' + id);
           return false;
         }