diff --git a/configuration/automations/bathroom-weight.yaml b/configuration/automations/bathroom-weight.yaml
index ce99c9d..cf85118 100644
--- a/configuration/automations/bathroom-weight.yaml
+++ b/configuration/automations/bathroom-weight.yaml
@@ -2,11 +2,9 @@
   alias: "[Bathroom] Weight measured"
   mode: single
   trigger:
-    - type: weight
-      platform: device
-      device_id: 6073ba909a68954d04c2ea86907a9193
-      entity_id: fcf113f550b430be77b620f80101c3a7
-      domain: sensor
+    - platform: state
+      entity_id:
+        - sensor.mi_smart_scale_6f96_mass
   condition: [ ]
   action:
     - service: notify.mobile_app_iphone