diff --git a/src/test/java/org/cloud/sonic/core/ios/IOSDriverTest.java b/src/test/java/org/cloud/sonic/core/ios/IOSDriverTest.java
index a0b387e..3ddad2b 100644
--- a/src/test/java/org/cloud/sonic/core/ios/IOSDriverTest.java
+++ b/src/test/java/org/cloud/sonic/core/ios/IOSDriverTest.java
@@ -34,11 +34,6 @@ public static Object[][] data() {
 
     @BeforeClass
     public static void before() throws InterruptedException, SonicRespException {
-        try {
-            iosDriver.lock();
-        } catch (SonicRespException e) {
-            Assert.assertEquals(e.getMessage(), "sessionId not found.");
-        }
         try {
             new IOSDriver("http://localhost:8100", null);
         } catch (SonicRespException e) {
@@ -110,5 +105,10 @@ public void testLock() throws SonicRespException {
     @AfterClass
     public static void after() throws SonicRespException {
         iosDriver.closeDriver();
+        try {
+            iosDriver.lock();
+        } catch (SonicRespException e) {
+            Assert.assertEquals(e.getMessage(), "sessionId not found.");
+        }
     }
 }