diff --git a/packages/js-dash-sdk/src/test/fixtures/createIdentityFixtureInAccount.ts b/packages/js-dash-sdk/src/test/fixtures/createIdentityFixtureInAccount.ts
index 1f19dc70964..537483b198e 100644
--- a/packages/js-dash-sdk/src/test/fixtures/createIdentityFixtureInAccount.ts
+++ b/packages/js-dash-sdk/src/test/fixtures/createIdentityFixtureInAccount.ts
@@ -15,8 +15,9 @@ export function createIdentityFixtureInAccount(account) {
         securityLevel: IdentityPublicKey.SECURITY_LEVELS.MASTER
     });
 
-    account.storage.insertIdentityIdAtIndex(
-        account.walletId,
+    account.storage
+      .getWalletStore(account.walletId)
+      .insertIdentityIdAtIndex(
         identityFixture.getId().toString(),
         identityFixtureIndex,
     );
diff --git a/packages/wallet-lib/tests/integration/plugins/Workers/TransactionSyncStreamWorker.spec.js b/packages/wallet-lib/tests/integration/plugins/Workers/TransactionSyncStreamWorker.spec.js
index c638e1efcd5..bee89ee4bb3 100644
--- a/packages/wallet-lib/tests/integration/plugins/Workers/TransactionSyncStreamWorker.spec.js
+++ b/packages/wallet-lib/tests/integration/plugins/Workers/TransactionSyncStreamWorker.spec.js
@@ -65,7 +65,7 @@ describe('TransactionSyncStreamWorker', function suite() {
     account = await wallet.getAccount();
 
     storage = account.storage;
-    walletId = Object.keys(storage.store.wallets)[0];
+    walletId = account.walletId;
 
     address = account.getAddress(0).address;
     addressAtIndex19 = account.getAddress(19).address;