diff --git a/custom_components/badnest/climate.py b/custom_components/badnest/climate.py
index e21f22d..b30d922 100644
--- a/custom_components/badnest/climate.py
+++ b/custom_components/badnest/climate.py
@@ -2,7 +2,7 @@
 from datetime import datetime
 import logging
 
-from homeassistant.components.climate import ClimateDevice
+from homeassistant.components.climate import ClimateEntity
 from homeassistant.components.climate.const import (
     ATTR_TARGET_TEMP_HIGH,
     ATTR_TARGET_TEMP_LOW,
@@ -78,8 +78,8 @@ async def async_setup_platform(hass,
     async_add_entities(thermostats)
 
 
-class NestClimate(ClimateDevice):
-    """Representation of a Nest climate device."""
+class NestClimate(ClimateEntity):
+    """Representation of a Nest climate entity."""
 
     def __init__(self, device_id, api):
         """Initialize the thermostat."""