diff --git a/esphome/components/fyrtur_motor/switch/__init__.py b/esphome/components/fyrtur_motor/switch/__init__.py index 3be3e6e217..ae2ec1e10c 100644 --- a/esphome/components/fyrtur_motor/switch/__init__.py +++ b/esphome/components/fyrtur_motor/switch/__init__.py @@ -2,7 +2,7 @@ import esphome.codegen as cg from esphome.components import switch import esphome.config_validation as cv from esphome.const import ( - DEVICE_CLASS_SWITCH, + ENTITY_CATEGORY_CONFIG, ICON_ARROW_EXPAND_VERTICAL, ) from .. import CONF_FYRTUR_MOTOR_ID, FyrturMotorComponent, fyrtur_motor_ns @@ -15,7 +15,7 @@ CONFIG_SCHEMA = { cv.GenerateID(CONF_FYRTUR_MOTOR_ID): cv.use_id(FyrturMotorComponent), cv.Required(CONF_OPEN_CLOSE): switch.switch_schema( OpenCloseSwitch, - device_class=DEVICE_CLASS_SWITCH, + entity_category=ENTITY_CATEGORY_CONFIG, icon=ICON_ARROW_EXPAND_VERTICAL, ), }