diff --git a/gateway/include/devices.h b/gateway/include/devices.h
index 494b156..ac68c03 100644
--- a/gateway/include/devices.h
+++ b/gateway/include/devices.h
@@ -41,12 +41,14 @@ namespace OilTank {
         return Builder<Sensor>::instance(new Sensor{ "Level", id })
             .asDevice(device)
             .withUnitMeasure("%")
+            .withSensorStateClass(MEASUREMENT)
             .withIcon("mdi:hydraulic-oil-level")
             .withValueTemplate("{{ 100 - ((value_json.sensor.value-12)|float*100/120)|round(2) }}")
             .addSecondary(
                 Builder<Sensor>::instance(new Sensor{ "Depth", id })
                 .withDeviceClass("distance")
                 .withUnitMeasure("cm")
+                .withSensorStateClass(MEASUREMENT)
                 .withValueTemplate("{{ value_json.sensor.value }}")
                 .build()
             )
diff --git a/gateway/platformio.ini b/gateway/platformio.ini
index c096e3f..6f8eb3f 100644
--- a/gateway/platformio.ini
+++ b/gateway/platformio.ini
@@ -33,7 +33,7 @@ framework = arduino
 lib_deps = 
 	${env.lib_deps}
 	arkhipenko/TaskScheduler@^3.8.5
-	https://git.hodos.ro/libraries/ha-mqtt.git@^1.6.0
+	https://git.hodos.ro/libraries/ha-mqtt.git@^1.7.0
 	https://git.hodos.ro/libraries/wifi.git@^2.0.0
 upload_port = 192.168.6.161
 upload_protocol = espota