From 1bc3c743db72eba8bb9d60ee4b0cdde17a949211 Mon Sep 17 00:00:00 2001
From: Franck Nijhof <git@frenck.dev>
Date: Thu, 5 Aug 2021 12:12:06 +0200
Subject: [PATCH] Add missing device class to SAJ energy sensors (#54048)

---
 homeassistant/components/saj/sensor.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/homeassistant/components/saj/sensor.py b/homeassistant/components/saj/sensor.py
index fb3b31764f8..1b46632051e 100644
--- a/homeassistant/components/saj/sensor.py
+++ b/homeassistant/components/saj/sensor.py
@@ -18,6 +18,7 @@ from homeassistant.const import (
     CONF_PASSWORD,
     CONF_TYPE,
     CONF_USERNAME,
+    DEVICE_CLASS_ENERGY,
     DEVICE_CLASS_POWER,
     DEVICE_CLASS_TEMPERATURE,
     ENERGY_KILO_WATT_HOUR,
@@ -204,6 +205,8 @@ class SAJsensor(SensorEntity):
         """Return the device class the sensor belongs to."""
         if self.unit_of_measurement == POWER_WATT:
             return DEVICE_CLASS_POWER
+        if self.unit_of_measurement == ENERGY_KILO_WATT_HOUR:
+            return DEVICE_CLASS_ENERGY
         if (
             self.unit_of_measurement == TEMP_CELSIUS
             or self._sensor.unit == TEMP_FAHRENHEIT