diff --git a/homeassistant/components/egardia/binary_sensor.py b/homeassistant/components/egardia/binary_sensor.py index 61f43301fd9..cf621277faf 100644 --- a/homeassistant/components/egardia/binary_sensor.py +++ b/homeassistant/components/egardia/binary_sensor.py @@ -1,7 +1,6 @@ """Interfaces with Egardia/Woonveilig alarm control panel.""" from homeassistant.components.binary_sensor import ( - DEVICE_CLASS_MOTION, - DEVICE_CLASS_OPENING, + BinarySensorDeviceClass, BinarySensorEntity, ) from homeassistant.const import STATE_OFF, STATE_ON @@ -9,9 +8,9 @@ from homeassistant.const import STATE_OFF, STATE_ON from . import ATTR_DISCOVER_DEVICES, EGARDIA_DEVICE EGARDIA_TYPE_TO_DEVICE_CLASS = { - "IR Sensor": DEVICE_CLASS_MOTION, - "Door Contact": DEVICE_CLASS_OPENING, - "IR": DEVICE_CLASS_MOTION, + "IR Sensor": BinarySensorDeviceClass.MOTION, + "Door Contact": BinarySensorDeviceClass.OPENING, + "IR": BinarySensorDeviceClass.MOTION, }