diff --git a/custom_components/xiaomi_miot/core/device_customizes.py b/custom_components/xiaomi_miot/core/device_customizes.py
index 640c8bd0b1..a006870894 100644
--- a/custom_components/xiaomi_miot/core/device_customizes.py
+++ b/custom_components/xiaomi_miot/core/device_customizes.py
@@ -2086,10 +2086,12 @@
         'number_properties': 'horizontal_angle,vertical_angle,off_delay',
     },
     'zhimi.heater.na1': {
-        'switch_properties': 'return_to_middle,alarm',
+        'switch_properties': 'on,return_to_middle',
         'number_properties': 'countdown_time',
     },
     'zhimi.heater.nb1': {
+        'switch_properties': 'on,return_to_middle',
+        'number_properties': 'countdown_time',
         'brightness_for_on': 0,
         'brightness_for_off': 2,
     },