diff --git a/custom_components/tuya_local/devices/polytherm_polyalpha_thermostat.yaml b/custom_components/tuya_local/devices/polytherm_polyalpha_thermostat.yaml
index 27b3ec0ba7..ba54fcc968 100644
--- a/custom_components/tuya_local/devices/polytherm_polyalpha_thermostat.yaml
+++ b/custom_components/tuya_local/devices/polytherm_polyalpha_thermostat.yaml
@@ -4,25 +4,22 @@ products:
     manufacturer: Polytherm
     model: Polyalpha
 entities:
-  - entity: switch
-    dps:
-      - id: 1
-        type: boolean
-        name: switch
   - entity: climate
     translation_key: thermostat
     dps:
       - id: 1
-        name: available
-        type: boolean
-      - id: 104
         name: hvac_mode
         type: boolean
         mapping:
           - dps_val: false
-            value: cool
+            value: "off"
           - dps_val: true
-            value: heat
+            constraint: cool_heat
+            conditions:
+              - dps_val: false
+                value: cool
+              - dps_val: true
+                value: heat
       - id: 2
         name: preset_mode
         type: string
@@ -51,7 +48,7 @@ entities:
         type: string
         mapping:
           - dps_val: open
-            constraint: hvac_mode
+            constraint: cool_heat
             conditions:
               - dps_val: false
                 value: cooling
@@ -59,23 +56,10 @@ entities:
                 value: heating
           - dps_val: close
             value: idle
-  - entity: humidifier
-    name: Humidity override
-    dps:
-      - id: 1
+      - id: 104
         type: boolean
-        name: available
-      - id: 106
-        type: integer
-        optional: true
-        name: humidity
-        range:
-          min: 50
-          max: 80
-      - id: 107
-        type: integer
-        optional: true
-        name: current_humidity
+        name: cool_heat
+        hidden: true
   - entity: lock
     translation_key: child_lock
     category: config
@@ -97,6 +81,7 @@ entities:
   - entity: sensor
     name: Floor temperature
     class: temperature
+    category: diagnostic
     dps:
       - id: 103
         type: integer