diff --git a/tests/e2e/tests/regression/tree-view/tree-view.e2e.ts b/tests/e2e/tests/regression/tree-view/tree-view.e2e.ts
index ffebc539b8..d2b0b37328 100644
--- a/tests/e2e/tests/regression/tree-view/tree-view.e2e.ts
+++ b/tests/e2e/tests/regression/tree-view/tree-view.e2e.ts
@@ -68,6 +68,8 @@ test('Verify that when user deletes the key he can see the key is removed from t
     await t.click(browserPage.treeViewDeviceFolder.nth(5));
     await browserPage.deleteKey();
     // Verify the results
+    await t.expect(browserPage.treeViewDeviceFolder.nth(2).exists).notOk('The previous folder is not closed after removing key folder');
+    await t.click(browserPage.treeViewDeviceFolder);
     await t.expect(browserPage.treeViewDeviceFolder.nth(2).textContent).notEql(keyFolder, 'The key folder is not removed from the tree view');
     await t.expect(browserPage.treeViewDeviceKyesCount.textContent).notEql(numberOfKeys, 'The number of keys is not recalculated');
 });