diff --git a/homeassistant/helpers/category_registry.py b/homeassistant/helpers/category_registry.py index ee0c8c1bb88..fec87262374 100644 --- a/homeassistant/helpers/category_registry.py +++ b/homeassistant/helpers/category_registry.py @@ -11,6 +11,7 @@ from homeassistant.core import HomeAssistant, callback from homeassistant.util.ulid import ulid_now from .registry import BaseRegistry +from .storage import Store from .typing import UNDEFINED, EventType, UndefinedType DATA_REGISTRY = "category_registry" @@ -46,7 +47,8 @@ class CategoryRegistry(BaseRegistry): """Initialize the category registry.""" self.hass = hass self.categories: dict[str, dict[str, CategoryEntry]] = {} - self._store = hass.helpers.storage.Store( + self._store: Store[dict[str, dict[str, list[dict[str, str]]]]] = Store( + hass, STORAGE_VERSION_MAJOR, STORAGE_KEY, atomic_writes=True,