From 5d1cf99756f3841cf5a39b9a52a1f2d1f05e1062 Mon Sep 17 00:00:00 2001 From: Aaron Bach Date: Fri, 8 Mar 2024 09:10:08 -0700 Subject: [PATCH] Fix tests --- homeassistant/components/notion/__init__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/homeassistant/components/notion/__init__.py b/homeassistant/components/notion/__init__.py index 33a749909282cb..2c1a6d0ac84237 100644 --- a/homeassistant/components/notion/__init__.py +++ b/homeassistant/components/notion/__init__.py @@ -181,7 +181,7 @@ def async_save_refresh_token(refresh_token: str) -> None: entry.async_on_unload(client.add_refresh_token_callback(async_save_refresh_token)) # Save the client's refresh token if it's different than what we already have: - if (token := client.refresh_token) and token != entry.data[CONF_REFRESH_TOKEN]: + if (token := client.refresh_token) and token != entry.data.get(CONF_REFRESH_TOKEN): async_save_refresh_token(token) hass.config_entries.async_update_entry(entry, **entry_updates)