Fix tests

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)