diff --git a/custom_components/blanco_unit/sensor.py b/custom_components/blanco_unit/sensor.py
index fe1efbb..b3c014a 100644
--- a/custom_components/blanco_unit/sensor.py
+++ b/custom_components/blanco_unit/sensor.py
@@ -86,7 +86,7 @@ class FilterRemainingSensor(BlancoUnitBaseEntity, SensorEntity):
 
     _attr_unique_id = "filter_remaining"
     _attr_translation_key = _attr_unique_id
-    _attr_device_class = SensorDeviceClass.BATTERY
+    _attr_icon = "mdi:history"
     _attr_native_unit_of_measurement = PERCENTAGE
     _attr_state_class = SensorStateClass.MEASUREMENT
 
@@ -108,7 +108,7 @@ class CO2RemainingSensor(BlancoUnitBaseEntity, SensorEntity):
 
     _attr_unique_id = "co2_remaining"
     _attr_translation_key = _attr_unique_id
-    _attr_device_class = SensorDeviceClass.BATTERY
+    _attr_icon = "mdi:history"
     _attr_native_unit_of_measurement = PERCENTAGE
     _attr_state_class = SensorStateClass.MEASUREMENT
 
diff --git a/tests/snapshots/test_sensor.ambr b/tests/snapshots/test_sensor.ambr
index 270cc98..b8ea8fc 100644
--- a/tests/snapshots/test_sensor.ambr
+++ b/tests/snapshots/test_sensor.ambr
@@ -122,8 +122,8 @@
     'name': None,
     'options': dict({
     }),
-    'original_device_class': <SensorDeviceClass.BATTERY: 'battery'>,
-    'original_icon': None,
+    'original_device_class': None,
+    'original_icon': 'mdi:history',
     'original_name': 'CO2 Cylinder Remaining',
     'platform': 'blanco_unit',
     'previous_unique_id': None,
@@ -137,8 +137,8 @@
 # name: test_all_entities[sensor.test_blanco_unit_co2_cylinder_remaining-state]
   StateSnapshot({
     'attributes': ReadOnlyDict({
-      'device_class': 'battery',
       'friendly_name': 'Test Blanco Unit CO2 Cylinder Remaining',
+      'icon': 'mdi:history',
       'state_class': <SensorStateClass.MEASUREMENT: 'measurement'>,
       'unit_of_measurement': '%',
     }),
@@ -469,8 +469,8 @@
     'name': None,
     'options': dict({
     }),
-    'original_device_class': <SensorDeviceClass.BATTERY: 'battery'>,
-    'original_icon': None,
+    'original_device_class': None,
+    'original_icon': 'mdi:history',
     'original_name': 'Filter Capacity Remaining',
     'platform': 'blanco_unit',
     'previous_unique_id': None,
@@ -484,8 +484,8 @@
 # name: test_all_entities[sensor.test_blanco_unit_filter_capacity_remaining-state]
   StateSnapshot({
     'attributes': ReadOnlyDict({
-      'device_class': 'battery',
       'friendly_name': 'Test Blanco Unit Filter Capacity Remaining',
+      'icon': 'mdi:history',
       'state_class': <SensorStateClass.MEASUREMENT: 'measurement'>,
       'unit_of_measurement': '%',
     }),