diff --git a/rule-engine/rule-engine-api/src/main/java/org/thingsboard/rule/engine/api/msg/DeviceAttributes.java b/rule-engine/rule-engine-api/src/main/java/org/thingsboard/rule/engine/api/msg/DeviceAttributes.java
index bb002a3068..0c8de63d27 100644
--- a/rule-engine/rule-engine-api/src/main/java/org/thingsboard/rule/engine/api/msg/DeviceAttributes.java
+++ b/rule-engine/rule-engine-api/src/main/java/org/thingsboard/rule/engine/api/msg/DeviceAttributes.java
@@ -71,7 +71,7 @@ public class DeviceAttributes {
     public void remove(AttributeKey key) {
         Map<String, AttributeKvEntry> map = getMapByScope(key.getScope());
         if (map != null) {
-            map.remove(key);
+            map.remove(key.getAttributeKey());
         }
     }