From c3cc566fe3e2ddce52c772bce5ae2d6125b7465f Mon Sep 17 00:00:00 2001
From: karwosts <32912880+karwosts@users.noreply.github.com>
Date: Sun, 1 Mar 2026 23:06:10 -0800
Subject: [PATCH] Fix distribution card stub error (#29915)

* Fix distribution card stub error

* unit check not required
---
 src/panels/lovelace/cards/hui-distribution-card.ts | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/panels/lovelace/cards/hui-distribution-card.ts b/src/panels/lovelace/cards/hui-distribution-card.ts
index 5b2e43e520..30cf512ed4 100644
--- a/src/panels/lovelace/cards/hui-distribution-card.ts
+++ b/src/panels/lovelace/cards/hui-distribution-card.ts
@@ -68,9 +68,9 @@ export class HuiDistributionCard
 
     // Strategy 1: Try to find power sensors (W, kW) - most common use case
     const powerFilter = (stateObj: HassEntity): boolean => {
-      const unit = stateObj.attributes.unit_of_measurement;
       const stateValue = Number(stateObj.state);
-      return (unit === "W" || unit === "kW") && !isNaN(stateValue);
+      const deviceClass = stateObj.attributes.device_class;
+      return deviceClass === "power" && !isNaN(stateValue);
     };
 
     let foundEntities = findEntities(