diff --git a/homeassistant/components/compensation/__init__.py b/homeassistant/components/compensation/__init__.py
index 7ab042ab5b30..54890de1605f 100644
--- a/homeassistant/components/compensation/__init__.py
+++ b/homeassistant/components/compensation/__init__.py
@@ -187,4 +187,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
 
 async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
     """Unload Compensation config entry."""
-    return await hass.config_entries.async_unload_platforms(entry, PLATFORMS)
+    unload_ok = await hass.config_entries.async_unload_platforms(entry, PLATFORMS)
+    if unload_ok:
+        hass.data[DATA_COMPENSATION].pop(entry.entry_id, None)
+    return unload_ok
diff --git a/homeassistant/components/compensation/config_flow.py b/homeassistant/components/compensation/config_flow.py
index 23e421e3b6cb..a37d810dda74 100644
--- a/homeassistant/components/compensation/config_flow.py
+++ b/homeassistant/components/compensation/config_flow.py
@@ -1,4 +1,4 @@
-"""Config flow for statistics."""
+"""Config flow for Compensation integration."""
 
 from __future__ import annotations
 
diff --git a/homeassistant/components/compensation/const.py b/homeassistant/components/compensation/const.py
index 6b8f0c7f1d0a..7e08202ba437 100644
--- a/homeassistant/components/compensation/const.py
+++ b/homeassistant/components/compensation/const.py
@@ -1,6 +1,9 @@
 """Compensation constants."""
 
+from typing import Any
+
 from homeassistant.const import Platform
+from homeassistant.util.hass_dict import HassKey
 
 DOMAIN = "compensation"
 PLATFORMS = [Platform.SENSOR]
@@ -16,7 +19,8 @@ CONF_PRECISION = "precision"
 CONF_POLYNOMIAL = "polynomial"
 CONF_POLYNOMIAL_CONFIG = "polynomial_config"
 
-DATA_COMPENSATION = "compensation_data"
+
+DATA_COMPENSATION: HassKey[dict[str, Any]] = HassKey("compensation_data")
 
 DEFAULT_DEGREE = 1
 DEFAULT_NAME = "Compensation"