diff --git a/homeassistant/components/axis/binary_sensor.py b/homeassistant/components/axis/binary_sensor.py index 7ef3838b1f7..01cfb834f26 100644 --- a/homeassistant/components/axis/binary_sensor.py +++ b/homeassistant/components/axis/binary_sensor.py @@ -17,10 +17,7 @@ from axis.event_stream import ( ) from homeassistant.components.binary_sensor import ( - DEVICE_CLASS_CONNECTIVITY, - DEVICE_CLASS_LIGHT, - DEVICE_CLASS_MOTION, - DEVICE_CLASS_SOUND, + BinarySensorDeviceClass, BinarySensorEntity, ) from homeassistant.core import callback @@ -32,10 +29,10 @@ from .axis_base import AxisEventBase from .const import DOMAIN as AXIS_DOMAIN DEVICE_CLASS = { - CLASS_INPUT: DEVICE_CLASS_CONNECTIVITY, - CLASS_LIGHT: DEVICE_CLASS_LIGHT, - CLASS_MOTION: DEVICE_CLASS_MOTION, - CLASS_SOUND: DEVICE_CLASS_SOUND, + CLASS_INPUT: BinarySensorDeviceClass.CONNECTIVITY, + CLASS_LIGHT: BinarySensorDeviceClass.LIGHT, + CLASS_MOTION: BinarySensorDeviceClass.MOTION, + CLASS_SOUND: BinarySensorDeviceClass.SOUND, }