diff --git a/homeassistant/components/myq/cover.py b/homeassistant/components/myq/cover.py index 861ae3790075..f3bc27a84537 100644 --- a/homeassistant/components/myq/cover.py +++ b/homeassistant/components/myq/cover.py @@ -3,9 +3,10 @@ from pymyq.const import DEVICE_TYPE_GATE as MYQ_DEVICE_TYPE_GATE from pymyq.errors import MyQError from homeassistant.components.cover import ( + SUPPORT_CLOSE, + SUPPORT_OPEN, CoverDeviceClass, CoverEntity, - CoverEntityFeature, ) from homeassistant.config_entries import ConfigEntry from homeassistant.const import STATE_CLOSED, STATE_CLOSING, STATE_OPEN, STATE_OPENING @@ -35,7 +36,7 @@ async def async_setup_entry( class MyQCover(MyQEntity, CoverEntity): """Representation of a MyQ cover.""" - _attr_supported_features = CoverEntityFeature.OPEN | CoverEntityFeature.CLOSE + _attr_supported_features = SUPPORT_OPEN | SUPPORT_CLOSE def __init__(self, coordinator, device): """Initialize with API object, device id."""