diff --git a/custom_components/hon/button.py b/custom_components/hon/button.py
index 1309914..934370d 100644
--- a/custom_components/hon/button.py
+++ b/custom_components/hon/button.py
@@ -4,7 +4,7 @@ import pkg_resources
 from homeassistant.components import persistent_notification
 from homeassistant.components.button import ButtonEntityDescription, ButtonEntity
 from homeassistant.config_entries import ConfigEntry
-from homeassistant.const import EntityCategory
+from homeassistant.helpers.entity import EntityCategory
 from pyhon.appliance import HonAppliance
 
 from .const import DOMAIN
diff --git a/custom_components/hon/switch.py b/custom_components/hon/switch.py
index 03273a7..7351a08 100644
--- a/custom_components/hon/switch.py
+++ b/custom_components/hon/switch.py
@@ -5,7 +5,7 @@ from typing import Any
 
 from homeassistant.components.switch import SwitchEntityDescription, SwitchEntity
 from homeassistant.config_entries import ConfigEntry
-from homeassistant.const import EntityCategory
+from homeassistant.helpers.entity import EntityCategory
 from homeassistant.core import callback
 from pyhon.parameter.base import HonParameter
 from pyhon.parameter.range import HonParameterRange