diff --git a/homeassistant/components/obihai/sensor.py b/homeassistant/components/obihai/sensor.py
index 7ba28ee0741..f54611283a6 100644
--- a/homeassistant/components/obihai/sensor.py
+++ b/homeassistant/components/obihai/sensor.py
@@ -5,13 +5,12 @@ import logging
 from pyobihai import PyObihai
 import voluptuous as vol
 
-from homeassistant.components.sensor import PLATFORM_SCHEMA, SensorEntity
-from homeassistant.const import (
-    CONF_HOST,
-    CONF_PASSWORD,
-    CONF_USERNAME,
-    DEVICE_CLASS_TIMESTAMP,
+from homeassistant.components.sensor import (
+    PLATFORM_SCHEMA,
+    SensorDeviceClass,
+    SensorEntity,
 )
+from homeassistant.const import CONF_HOST, CONF_PASSWORD, CONF_USERNAME
 import homeassistant.helpers.config_validation as cv
 
 _LOGGER = logging.getLogger(__name__)
@@ -105,7 +104,7 @@ class ObihaiServiceSensors(SensorEntity):
     def device_class(self):
         """Return the device class for uptime sensor."""
         if self._service_name == "Last Reboot":
-            return DEVICE_CLASS_TIMESTAMP
+            return SensorDeviceClass.TIMESTAMP
         return None
 
     @property