diff --git a/homeassistant/components/ted5000/sensor.py b/homeassistant/components/ted5000/sensor.py
index a7162ee9c63..c6aedbb5046 100644
--- a/homeassistant/components/ted5000/sensor.py
+++ b/homeassistant/components/ted5000/sensor.py
@@ -9,8 +9,8 @@ import xmltodict
 
 from homeassistant.components.sensor import (
     PLATFORM_SCHEMA,
-    STATE_CLASS_MEASUREMENT,
     SensorEntity,
+    SensorStateClass,
 )
 from homeassistant.const import (
     CONF_HOST,
@@ -62,7 +62,7 @@ def setup_platform(hass, config, add_entities, discovery_info=None):
 class Ted5000Sensor(SensorEntity):
     """Implementation of a Ted5000 sensor."""
 
-    _attr_state_class = STATE_CLASS_MEASUREMENT
+    _attr_state_class = SensorStateClass.MEASUREMENT
 
     def __init__(self, gateway, name, mtu, unit):
         """Initialize the sensor."""