diff --git a/application/src/test/java/org/thingsboard/server/service/state/DefaultDeviceStateServiceTest.java b/application/src/test/java/org/thingsboard/server/service/state/DefaultDeviceStateServiceTest.java
index 06cde65c36..746da9a8f0 100644
--- a/application/src/test/java/org/thingsboard/server/service/state/DefaultDeviceStateServiceTest.java
+++ b/application/src/test/java/org/thingsboard/server/service/state/DefaultDeviceStateServiceTest.java
@@ -363,7 +363,7 @@ public class DefaultDeviceStateServiceTest {
         service.deviceStates.put(deviceId, deviceStateDataMock);
 
         // WHEN
-        service.updateActivityState(deviceId, deviceStateDataMock, System.currentTimeMillis());
+        service.updateActivityState(deviceId, null, System.currentTimeMillis());
 
         // THEN
         assertThat(service.deviceStates.get(deviceId)).isNull();