diff --git a/config/water-meter.yaml b/config/water-meter.yaml
index 9e2dd46..f401d26 100644
--- a/config/water-meter.yaml
+++ b/config/water-meter.yaml
@@ -34,7 +34,7 @@
     name: "Water Flow Rate"
     unit_of_measurement: "gal/min"
     accuracy_decimals: 2
-    device_class: water
+    device_class: volume_flow_rate
     state_class: measurement
 
     internal_filter: 100ms