diff --git a/custom_components/thermal_vision/camera.py b/custom_components/thermal_vision/camera.py
index ebacb05..3f262e0 100644
--- a/custom_components/thermal_vision/camera.py
+++ b/custom_components/thermal_vision/camera.py
@@ -25,8 +25,7 @@
     CONF_HOST,
     CONF_NAME,
     CONF_VERIFY_SSL,
-    TEMP_CELSIUS,
-    TEMP_FAHRENHEIT,
+    UnitOfTemperature,
     STATE_UNKNOWN,
     STATE_UNAVAILABLE,
 )
@@ -209,24 +208,24 @@ def extra_state_attributes(self):
         return {
             "fps": self._fps,
             "min": self._pixel_min_temp
-            if self._temperature_unit == TEMP_CELSIUS
+            if self._temperature_unit == UnitOfTemperature.CELSIUS
             else TemperatureConverter.convert(
-                self._pixel_min_temp, TEMP_CELSIUS, TEMP_FAHRENHEIT
+                self._pixel_min_temp, UnitOfTemperature.CELSIUS, UnitOfTemperature.FAHRENHEIT
             ),
             "max": self._pixel_max_temp
-            if self._temperature_unit == TEMP_CELSIUS
+            if self._temperature_unit == UnitOfTemperature.CELSIUS
             else TemperatureConverter.convert(
-                self._pixel_max_temp, TEMP_CELSIUS, TEMP_FAHRENHEIT
+                self._pixel_max_temp, UnitOfTemperature.CELSIUS, UnitOfTemperature.FAHRENHEIT
             ),
             "range_min": self._min_temperature
-            if self._temperature_unit == TEMP_CELSIUS
+            if self._temperature_unit == UnitOfTemperature.CELSIUS
             else TemperatureConverter.convert(
-                self._min_temperature, TEMP_CELSIUS, TEMP_FAHRENHEIT
+                self._min_temperature, UnitOfTemperature.CELSIUS, UnitOfTemperature.FAHRENHEIT
             ),
             "range_max": self._max_temperature
-            if self._temperature_unit == TEMP_CELSIUS
+            if self._temperature_unit == UnitOfTemperature.CELSIUS
             else TemperatureConverter.convert(
-                self._max_temperature, TEMP_CELSIUS, TEMP_FAHRENHEIT
+                self._max_temperature, UnitOfTemperature.CELSIUS, UnitOfTemperature.FAHRENHEIT
             ),
         }
 
@@ -393,30 +392,30 @@ def _camera_image(self, pixels):
         if self._overlay:
             min_temp = (
                 self._pixel_min_temp
-                if self._temperature_unit == TEMP_CELSIUS
+                if self._temperature_unit == UnitOfTemperature.CELSIUS
                 else TemperatureConverter.convert(
-                    self._pixel_min_temp, TEMP_CELSIUS, TEMP_FAHRENHEIT
+                    self._pixel_min_temp, UnitOfTemperature.CELSIUS, UnitOfTemperature.FAHRENHEIT
                 )
             )
             max_temp = (
                 self._pixel_max_temp
-                if self._temperature_unit == TEMP_CELSIUS
+                if self._temperature_unit == UnitOfTemperature.CELSIUS
                 else TemperatureConverter.convert(
-                    self._pixel_max_temp, TEMP_CELSIUS, TEMP_FAHRENHEIT
+                    self._pixel_max_temp, UnitOfTemperature.CELSIUS, UnitOfTemperature.FAHRENHEIT
                 )
             )
             min_temperature = (
                 self._min_temperature
-                if self._temperature_unit == TEMP_CELSIUS
+                if self._temperature_unit == UnitOfTemperature.CELSIUS
                 else TemperatureConverter.convert(
-                    self._min_temperature, TEMP_CELSIUS, TEMP_FAHRENHEIT
+                    self._min_temperature, UnitOfTemperature.CELSIUS, UnitOfTemperature.FAHRENHEIT
                 )
             )
             max_temperature = (
                 self._max_temperature
-                if self._temperature_unit == TEMP_CELSIUS
+                if self._temperature_unit == UnitOfTemperature.CELSIUS
                 else TemperatureConverter.convert(
-                    self._max_temperature, TEMP_CELSIUS, TEMP_FAHRENHEIT
+                    self._max_temperature, UnitOfTemperature.CELSIUS, UnitOfTemperature.FAHRENHEIT
                 )
             )
 
diff --git a/custom_components/thermal_vision/sensor.py b/custom_components/thermal_vision/sensor.py
index 2ebf0a8..575fae7 100644
--- a/custom_components/thermal_vision/sensor.py
+++ b/custom_components/thermal_vision/sensor.py
@@ -21,7 +21,7 @@
     CONF_HOST,
     CONF_SCAN_INTERVAL,
     CONF_VERIFY_SSL,
-    TEMP_CELSIUS,
+    UnitOfTemperature,
 )
 
 from .const import (
@@ -180,7 +180,7 @@ def native_unit_of_measurement(self):
         if self._state_type == ATTR_PERSON_DETECTED:
             return None
         else:
-            return TEMP_CELSIUS
+            return UnitOfTemperature.CELSIUS
 
     @property
     def device_class(self):