diff --git a/lib/extension/homeassistant.js b/lib/extension/homeassistant.js
index fb49d006f..d37b26b8f 100644
--- a/lib/extension/homeassistant.js
+++ b/lib/extension/homeassistant.js
@@ -583,7 +583,6 @@ const mapping = {
     'HS1-WL-E': [configurations.binary_sensor_water_leak],
     '421786': [configurations.light_brightness],
     'ICZB-IW11D': [configurations.light_brightness],
-    'HGZB-01A': [configurations.light_brightness],
     '3321-S': [configurations.binary_sensor_contact, configurations.sensor_temperature],
     'ZPIR-8000': [configurations.binary_sensor_occupancy],
     'ZCTS-808': [configurations.binary_sensor_contact],
@@ -712,7 +711,8 @@ const mapping = {
     'HGZB-1S': [configurations.switch, configurations.sensor_click],
     'HGZB-045': [configurations.switch, configurations.sensor_click],
     'HGZB-43': [switchWithPostfix('top'), switchWithPostfix('bottom'), switchWithPostfix('center')],
-    'HGZB-01A/02A': [configurations.switch],
+    'HGZB-01A': [configurations.switch],
+    'HGZB-02A': [configurations.light_brightness],
     'MCT-350 SMA': [configurations.binary_sensor_contact],
     '3310-S': [configurations.sensor_temperature, configurations.sensor_battery],
     'IM6001-WLP01': [