diff --git a/lib/devices.js b/lib/devices.js
index 1daf3009..12a4cde9 100644
--- a/lib/devices.js
+++ b/lib/devices.js
@@ -105,6 +105,7 @@ const devices = {
         vendor: 'Xiaomi',
         description: 'Aqara wireless switch',
         supports: 'single, double, triple, quadruple click',
+        homeassistant: [homeassistant.sensor_button]
     },
     'lumi.sensor_86sw2\u0000Un': {
         model: 'WXKG02LM',