diff --git a/homeassistant/components/supla/cover.py b/homeassistant/components/supla/cover.py index c76b28e9de0..0c6715e3728 100644 --- a/homeassistant/components/supla/cover.py +++ b/homeassistant/components/supla/cover.py @@ -2,11 +2,7 @@ import logging from pprint import pformat -from homeassistant.components.cover import ( - ATTR_POSITION, - DEVICE_CLASS_GARAGE, - CoverEntity, -) +from homeassistant.components.cover import ATTR_POSITION, CoverDeviceClass, CoverEntity from . import DOMAIN, SUPLA_COORDINATORS, SUPLA_SERVERS, SuplaChannel @@ -115,4 +111,4 @@ class SuplaGateDoor(SuplaChannel, CoverEntity): @property def device_class(self): """Return the class of this device, from component DEVICE_CLASSES.""" - return DEVICE_CLASS_GARAGE + return CoverDeviceClass.GARAGE