diff --git a/modules/integration/tests-integration/tests-backend/src/test/java/org/wso2/identity/integration/test/identity/mgt/AccountLockEnabledTestCase.java b/modules/integration/tests-integration/tests-backend/src/test/java/org/wso2/identity/integration/test/identity/mgt/AccountLockEnabledTestCase.java
index 63e39eeed24..830463b40e9 100644
--- a/modules/integration/tests-integration/tests-backend/src/test/java/org/wso2/identity/integration/test/identity/mgt/AccountLockEnabledTestCase.java
+++ b/modules/integration/tests-integration/tests-backend/src/test/java/org/wso2/identity/integration/test/identity/mgt/AccountLockEnabledTestCase.java
@@ -138,11 +138,10 @@ public void testSuccessfulEmailRetrieval() {
     @AfterClass(alwaysRun = true)
     public void atEnd() throws Exception {
 
-        File identityMgtConfigFile = new File(getISResourceLocation()
-                + File.separator + "identityMgt" + File.separator
-                + "identity-accountlock-enabled.xml");
+        usmClient.deleteUser(testLockUser1);
+        usmClient.deleteUser(testLockUser2);
 
-        scm.applyConfigurationWithoutRestart(identityMgtConfigFile, identityXmlServerFile, true);
+        scm.restoreToLastConfiguration();
         scm.restartGracefully();
 
     }