diff --git a/custom_components/climate_group/climate.py b/custom_components/climate_group/climate.py
index b65a460..bd1d925 100755
--- a/custom_components/climate_group/climate.py
+++ b/custom_components/climate_group/climate.py
@@ -299,11 +299,11 @@ async def async_set_temperature(self, **kwargs: Any) -> None:
             await self.async_set_hvac_mode(kwargs[ATTR_HVAC_MODE])
 
         if ATTR_TEMPERATURE in kwargs:
-            data["target_temperature"] = kwargs[ATTR_TEMPERATURE]
+            data[ATTR_TEMPERATURE] = kwargs[ATTR_TEMPERATURE]
         if ATTR_TARGET_TEMP_LOW in kwargs:
-            data["target_temperature_low"] = kwargs[ATTR_TARGET_TEMP_LOW]
+            data[ATTR_TARGET_TEMP_LOW] = kwargs[ATTR_TARGET_TEMP_LOW]
         if ATTR_TARGET_TEMP_HIGH in kwargs:
-            data["target_temperature_high"] = kwargs[ATTR_TARGET_TEMP_HIGH]
+            data[ATTR_TARGET_TEMP_HIGH] = kwargs[ATTR_TARGET_TEMP_HIGH]
 
         _LOGGER.debug("Setting temperature: %s", data)