diff --git a/devices/sonoff-s31-reset-only.yaml b/devices/sonoff-s31-reset-only.yaml
index c9b2d85..1a612ee 100644
--- a/devices/sonoff-s31-reset-only.yaml
+++ b/devices/sonoff-s31-reset-only.yaml
@@ -77,7 +77,7 @@ sensor:
     energy: # total (active) energy value of the sensor in Wh
       name: "Energy"
       accuracy_decimals: 3 # Default
-      unit_of_measurement: "kWh" # Default
+      unit_of_measurement: "Wh" # Default
       device_class: energy # Default
       state_class: total_increasing # Default
       filters:
diff --git a/devices/sonoff-s31-reset-relay-toggle.yaml b/devices/sonoff-s31-reset-relay-toggle.yaml
index 0058c30..dc92be9 100644
--- a/devices/sonoff-s31-reset-relay-toggle.yaml
+++ b/devices/sonoff-s31-reset-relay-toggle.yaml
@@ -77,7 +77,7 @@ sensor:
     energy: # total (active) energy value of the sensor in Wh
       name: "Energy"
       accuracy_decimals: 3 # Default
-      unit_of_measurement: "kWh" # Default
+      unit_of_measurement: "Wh" # Default
       device_class: energy # Default
       state_class: total_increasing # Default
       filters: