From 4160f9c8944194cb5cb281768d302c072fb043b5 Mon Sep 17 00:00:00 2001
From: Guy Badman <61918526+badguy99@users.noreply.github.com>
Date: Wed, 20 May 2020 20:25:37 +0100
Subject: [PATCH] change from deprecated device to entity

---
 custom_components/badnest/water_heater.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/custom_components/badnest/water_heater.py b/custom_components/badnest/water_heater.py
index 1fa0e6c..ca1916a 100644
--- a/custom_components/badnest/water_heater.py
+++ b/custom_components/badnest/water_heater.py
@@ -96,7 +96,7 @@ async def async_setup_platform(hass,
     )
 
 
-class NestWaterHeater(WaterHeaterDevice):
+class NestWaterHeater(WaterHeaterEntity):
     """Representation of a Nest water heater device."""
 
     def __init__(self, device_id, api):