diff --git a/homeassistant/components/integration/sensor.py b/homeassistant/components/integration/sensor.py
index cd3e376b792..d9a41cf714e 100644
--- a/homeassistant/components/integration/sensor.py
+++ b/homeassistant/components/integration/sensor.py
@@ -8,7 +8,7 @@ from homeassistant.components.sensor import (
     DEVICE_CLASS_ENERGY,
     DEVICE_CLASS_POWER,
     PLATFORM_SCHEMA,
-    STATE_CLASS_TOTAL_INCREASING,
+    STATE_CLASS_TOTAL,
     SensorEntity,
 )
 from homeassistant.const import (
@@ -116,7 +116,7 @@ class IntegrationSensor(RestoreEntity, SensorEntity):
         self._unit_of_measurement = unit_of_measurement
         self._unit_prefix = UNIT_PREFIXES[unit_prefix]
         self._unit_time = UNIT_TIME[unit_time]
-        self._attr_state_class = STATE_CLASS_TOTAL_INCREASING
+        self._attr_state_class = STATE_CLASS_TOTAL
 
     async def async_added_to_hass(self):
         """Handle entity which will be added."""
diff --git a/tests/components/integration/test_sensor.py b/tests/components/integration/test_sensor.py
index e8aaf906936..5eff62835ba 100644
--- a/tests/components/integration/test_sensor.py
+++ b/tests/components/integration/test_sensor.py
@@ -2,7 +2,7 @@
 from datetime import timedelta
 from unittest.mock import patch
 
-from homeassistant.components.sensor import STATE_CLASS_TOTAL_INCREASING
+from homeassistant.components.sensor import STATE_CLASS_TOTAL
 from homeassistant.const import (
     DEVICE_CLASS_ENERGY,
     DEVICE_CLASS_POWER,
@@ -39,7 +39,7 @@ async def test_state(hass) -> None:
 
     state = hass.states.get("sensor.integration")
     assert state is not None
-    assert state.attributes.get("state_class") == STATE_CLASS_TOTAL_INCREASING
+    assert state.attributes.get("state_class") == STATE_CLASS_TOTAL
     assert "device_class" not in state.attributes
 
     future_now = dt_util.utcnow() + timedelta(seconds=3600)
@@ -57,7 +57,7 @@ async def test_state(hass) -> None:
 
     assert state.attributes.get("unit_of_measurement") == ENERGY_KILO_WATT_HOUR
     assert state.attributes.get("device_class") == DEVICE_CLASS_ENERGY
-    assert state.attributes.get("state_class") == STATE_CLASS_TOTAL_INCREASING
+    assert state.attributes.get("state_class") == STATE_CLASS_TOTAL
 
 
 async def test_restore_state(hass: HomeAssistant) -> None:
@@ -104,7 +104,9 @@ async def test_restore_state_failed(hass: HomeAssistant) -> None:
             State(
                 "sensor.integration",
                 "INVALID",
-                {},
+                {
+                    "last_reset": "2019-10-06T21:00:00.000000",
+                },
             ),
         ),
     )
@@ -125,7 +127,7 @@ async def test_restore_state_failed(hass: HomeAssistant) -> None:
     assert state
     assert state.state == "0"
     assert state.attributes.get("unit_of_measurement") == ENERGY_KILO_WATT_HOUR
-    assert state.attributes.get("state_class") == STATE_CLASS_TOTAL_INCREASING
+    assert state.attributes.get("state_class") == STATE_CLASS_TOTAL
     assert "device_class" not in state.attributes