diff --git a/custom_components/alexa_media/sensor.py b/custom_components/alexa_media/sensor.py
index 5d05ccc3..cdfd195c 100644
--- a/custom_components/alexa_media/sensor.py
+++ b/custom_components/alexa_media/sensor.py
@@ -206,15 +206,19 @@ def device_info(self):
         return None
 
     @property
-    def unit_of_measurement(self):
+    def native_unit_of_measurement(self):
         return TEMP_CELSIUS
 
     @property
-    def state(self):
+    def native_value(self):
         return parse_temperature_from_coordinator(
             self.coordinator, self.alexa_entity_id
         )
 
+    @property
+    def device_class(self):
+        return "temperature"
+
     @property
     def unique_id(self):
         # This includes "_temperature" because the Alexa entityId is for a physical device