diff --git a/homeassistant/components/waterfurnace/sensor.py b/homeassistant/components/waterfurnace/sensor.py
index 5d7832ca58d..1e543d15c60 100644
--- a/homeassistant/components/waterfurnace/sensor.py
+++ b/homeassistant/components/waterfurnace/sensor.py
@@ -1,12 +1,11 @@
 """Support for Waterfurnace."""
 
-from homeassistant.components.sensor import ENTITY_ID_FORMAT, SensorEntity
-from homeassistant.const import (
-    DEVICE_CLASS_TEMPERATURE,
-    PERCENTAGE,
-    POWER_WATT,
-    TEMP_FAHRENHEIT,
+from homeassistant.components.sensor import (
+    ENTITY_ID_FORMAT,
+    SensorDeviceClass,
+    SensorEntity,
 )
+from homeassistant.const import PERCENTAGE, POWER_WATT, TEMP_FAHRENHEIT
 from homeassistant.core import callback
 from homeassistant.util import slugify
 
@@ -40,13 +39,21 @@ SENSORS = [
         "tstatactivesetpoint",
         None,
         TEMP_FAHRENHEIT,
-        DEVICE_CLASS_TEMPERATURE,
+        SensorDeviceClass.TEMPERATURE,
     ),
     WFSensorConfig(
-        "Leaving Air", "leavingairtemp", None, TEMP_FAHRENHEIT, DEVICE_CLASS_TEMPERATURE
+        "Leaving Air",
+        "leavingairtemp",
+        None,
+        TEMP_FAHRENHEIT,
+        SensorDeviceClass.TEMPERATURE,
     ),
     WFSensorConfig(
-        "Room Temp", "tstatroomtemp", None, TEMP_FAHRENHEIT, DEVICE_CLASS_TEMPERATURE
+        "Room Temp",
+        "tstatroomtemp",
+        None,
+        TEMP_FAHRENHEIT,
+        SensorDeviceClass.TEMPERATURE,
     ),
     WFSensorConfig("Loop Temp", "enteringwatertemp", None, TEMP_FAHRENHEIT),
     WFSensorConfig(