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(