diff --git a/package-lock.json b/package-lock.json
index e292ed1..6525bbc 100644
--- a/package-lock.json
+++ b/package-lock.json
@@ -9,7 +9,7 @@
       "version": "1.0.0",
       "license": "ISC",
       "dependencies": {
-        "@0xpolygonid/js-sdk": "^1.0.0",
+        "@0xpolygonid/js-sdk": "https://github.com/0xPolygonID/js-sdk#feature/pid-1037",
         "@iden3/js-jwz": "^1.0.0",
         "dotenv": "^16.3.1",
         "ethers": "^5.7.2"
@@ -20,9 +20,10 @@
       }
     },
     "node_modules/@0xpolygonid/js-sdk": {
-      "version": "1.0.0",
-      "resolved": "https://registry.npmjs.org/@0xpolygonid/js-sdk/-/js-sdk-1.0.0.tgz",
-      "integrity": "sha512-d09ps+w9jv5JHyezXLsao3MOiyUMHiprpv9WIiLzeudjD2quT/AnIwQpfxe8VK34QuEXaCiuP4KYbDDE4uXWDA==",
+      "version": "1.0.1",
+      "resolved": "git+ssh://git@github.com/0xPolygonID/js-sdk.git#106e71383bfefc686ff097af37c09bd8c0501dac",
+      "integrity": "sha512-VaXzjbhQ/rK8DKD75D87Q1L+4So0eQZ4eqpsHxBn1RgJBbshjRuCcmL4vsasfLQBM8uHMC1YGT4ej1OW5Ed+Ng==",
+      "license": "AGPL-3.0",
       "dependencies": {
         "@iden3/js-crypto": "1.0.0",
         "@iden3/js-iden3-core": "1.0.0",
@@ -2493,9 +2494,9 @@
       }
     },
     "node_modules/undici": {
-      "version": "5.22.1",
-      "resolved": "https://registry.npmjs.org/undici/-/undici-5.22.1.tgz",
-      "integrity": "sha512-Ji2IJhFXZY0x/0tVBXeQwgPlLWw13GVzpsWPQ3rV50IFMMof2I55PZZxtm4P6iNq+L5znYN9nSTAq0ZyE6lSJw==",
+      "version": "5.23.0",
+      "resolved": "https://registry.npmjs.org/undici/-/undici-5.23.0.tgz",
+      "integrity": "sha512-1D7w+fvRsqlQ9GscLBwcAJinqcZGHUKjbOmXdlE/v8BvEGXjeWAax+341q44EuTcHXXnfyKNbKRq4Lg7OzhMmg==",
       "dependencies": {
         "busboy": "^1.6.0"
       },
diff --git a/package.json b/package.json
index 74f8e35..dd8bc4c 100644
--- a/package.json
+++ b/package.json
@@ -14,7 +14,7 @@
   "author": "",
   "license": "ISC",
   "dependencies": {
-    "@0xpolygonid/js-sdk": "^1.0.0",
+    "@0xpolygonid/js-sdk": "https://github.com/0xPolygonID/js-sdk#feature/pid-1037",
     "@iden3/js-jwz": "^1.0.0",
     "dotenv": "^16.3.1",
     "ethers": "^5.7.2"
diff --git a/walletSetup.ts b/walletSetup.ts
index ef30141..e50b17a 100644
--- a/walletSetup.ts
+++ b/walletSetup.ts
@@ -15,6 +15,7 @@ import {
   InMemoryDataSource,
   InMemoryMerkleTreeStorage,
   InMemoryPrivateKeyStore,
+  EncryptedKeyStore,
   KMS,
   KmsKeyType,
   Profile,
@@ -73,7 +74,9 @@ export async function initIdentityWallet(
   dataStorage: IDataStorage,
   credentialWallet: ICredentialWallet
 ): Promise<IIdentityWallet> {
-  const memoryKeyStore = new InMemoryPrivateKeyStore();
+  const memoryKeyStore = new EncryptedKeyStore<InMemoryPrivateKeyStore>(InMemoryPrivateKeyStore, {
+    password: 'p@ssword1'
+  });
   const bjjProvider = new BjjProvider(KmsKeyType.BabyJubJub, memoryKeyStore);
   const kms = new KMS();
   kms.registerKeyProvider(KmsKeyType.BabyJubJub, bjjProvider);