From 4a788b62d1be8b946a69f7276f2669444dd60a2a Mon Sep 17 00:00:00 2001 From: "J. Nick Koston" Date: Mon, 20 Apr 2020 13:21:00 -0500 Subject: [PATCH] Switch griddy to async_on_remove (#34471) --- homeassistant/components/griddy/sensor.py | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) diff --git a/homeassistant/components/griddy/sensor.py b/homeassistant/components/griddy/sensor.py index b072c613d41..9a58e5e3286 100644 --- a/homeassistant/components/griddy/sensor.py +++ b/homeassistant/components/griddy/sensor.py @@ -70,8 +70,6 @@ class GriddyPriceSensor(Entity): async def async_added_to_hass(self): """Subscribe to updates.""" - self._coordinator.async_add_listener(self.async_write_ha_state) - - async def async_will_remove_from_hass(self): - """Undo subscription.""" - self._coordinator.async_remove_listener(self.async_write_ha_state) + self.async_on_remove( + self._coordinator.async_add_listener(self.async_write_ha_state) + )