diff --git a/application/src/test/java/org/thingsboard/server/edge/imitator/EdgeImitator.java b/application/src/test/java/org/thingsboard/server/edge/imitator/EdgeImitator.java
index 69470a681d..e1b54b6f84 100644
--- a/application/src/test/java/org/thingsboard/server/edge/imitator/EdgeImitator.java
+++ b/application/src/test/java/org/thingsboard/server/edge/imitator/EdgeImitator.java
@@ -80,7 +80,7 @@ public class EdgeImitator {
     }
 
     public void disconnect() throws InterruptedException {
-        edgeRpcClient.disconnect();
+        edgeRpcClient.disconnect(false);
     }
 
     private void onUplinkResponse(UplinkResponseMsg msg) {