diff --git a/agent/src/main/java/com/microsoft/hydralab/agent/runner/TestRunner.java b/agent/src/main/java/com/microsoft/hydralab/agent/runner/TestRunner.java
index 35ed6dfab..7185131a6 100644
--- a/agent/src/main/java/com/microsoft/hydralab/agent/runner/TestRunner.java
+++ b/agent/src/main/java/com/microsoft/hydralab/agent/runner/TestRunner.java
@@ -50,6 +50,7 @@ public void runTestOnDevice(TestTask testTask, DeviceInfo deviceInfo, Logger log
         logger.info("Start running tests {}, timeout {}s", testTask.getTestSuite(), testTask.getTimeOutSecond());
 
         TestRun testRun = createTestRun(deviceInfo, testTask, logger);
+        testDeviceManager = deviceInfo.getTestDeviceManager();
         checkTestTaskCancel(testTask);
 
         try {