diff --git a/config/water-meter.yaml b/config/water-meter.yaml
index c4270c3..9e2dd46 100644
--- a/config/water-meter.yaml
+++ b/config/water-meter.yaml
@@ -44,6 +44,7 @@
       name: "Water Usage Total"
       unit_of_measurement: "gal"
       accuracy_decimals: 0
+      device_class: water
       state_class: total_increasing
 
 wifi: