From 927c2314c4e7cac55f9ac0cf92a109c5147ea9fe Mon Sep 17 00:00:00 2001 From: Chris Talkington Date: Sat, 4 Apr 2020 21:42:51 -0500 Subject: [PATCH] use async_on_remove for IPP Update Coordinator (#33596) --- homeassistant/components/ipp/__init__.py | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) diff --git a/homeassistant/components/ipp/__init__.py b/homeassistant/components/ipp/__init__.py index 447665a3676..81d77b27a3e 100644 --- a/homeassistant/components/ipp/__init__.py +++ b/homeassistant/components/ipp/__init__.py @@ -168,11 +168,9 @@ class IPPEntity(Entity): async def async_added_to_hass(self) -> None: """Connect to dispatcher listening for entity data notifications.""" - self.coordinator.async_add_listener(self.async_write_ha_state) - - async def async_will_remove_from_hass(self) -> None: - """Disconnect from update signal.""" - self.coordinator.async_remove_listener(self.async_write_ha_state) + self.async_on_remove( + self.coordinator.async_add_listener(self.async_write_ha_state) + ) async def async_update(self) -> None: """Update an IPP entity."""