diff --git a/custom_components/xiaomi_miot/core/device_customizes.py b/custom_components/xiaomi_miot/core/device_customizes.py
index 8255eba91..c10763565 100644
--- a/custom_components/xiaomi_miot/core/device_customizes.py
+++ b/custom_components/xiaomi_miot/core/device_customizes.py
@@ -1711,6 +1711,15 @@
         'device_class': 'energy',
         'unit_of_measurement': 'kWh',
     },
+    'zimi.waterheater.h01': {
+        'sensor_properties': 'water_heater.status,water_level,temperature_tips,'
+                             'input_water_temp,water_pump_volt,remaining_time',
+        'switch_properties': 'on,hot_water_recirculation,heat_preservation,jog_function,pressure_boost.status,'
+                             'cloud_preheater,in_cloud_timer,water_trigger,reminder,enhanced',
+        'select_properties': 'effective,model',
+        'number_properties': 'target_temperature,light_off_time,children_mode_temp,comfort_mode_temp,adult_mode_temp,'
+                             'kitchen_mode_temp,pet_mode_temp,custom_time,boost_value',
+    },
     'zinguo.switch.b5m': {
         'main_miot_services': 'switch-2',
         'sensor_properties': 'temperature',