diff --git a/homeassistant/components/digital_ocean/binary_sensor.py b/homeassistant/components/digital_ocean/binary_sensor.py index af74e1ffb13..f2222a03d73 100644 --- a/homeassistant/components/digital_ocean/binary_sensor.py +++ b/homeassistant/components/digital_ocean/binary_sensor.py @@ -4,8 +4,8 @@ import logging import voluptuous as vol from homeassistant.components.binary_sensor import ( - DEVICE_CLASS_MOVING, PLATFORM_SCHEMA, + BinarySensorDeviceClass, BinarySensorEntity, ) from homeassistant.const import ATTR_ATTRIBUTION @@ -74,7 +74,7 @@ class DigitalOceanBinarySensor(BinarySensorEntity): @property def device_class(self): """Return the class of this sensor.""" - return DEVICE_CLASS_MOVING + return BinarySensorDeviceClass.MOVING @property def extra_state_attributes(self):