diff --git a/src/App.tsx b/src/App.tsx
index 86cf663..f1cdb87 100644
--- a/src/App.tsx
+++ b/src/App.tsx
@@ -607,7 +607,7 @@ function App() {
     let wallet = structuredClone(rawWallet);
 
     const res = await decryptStoredWallet(password, wallet);
-    const wallet2 = new PhraseWallet(res, walletVersion);
+    const wallet2 = new PhraseWallet(res, wallet?.version || walletVersion);
     wallet = await wallet2.generateSaveWalletData(
       password,
       crypto.kdfThreads,
diff --git a/src/background-cases.ts b/src/background-cases.ts
index 0eaed21..98aabf4 100644
--- a/src/background-cases.ts
+++ b/src/background-cases.ts
@@ -232,7 +232,8 @@ export async function userInfoCase(request, event) {
 export async function decryptWalletCase(request, event) {
   try { 
     const { password, wallet } = request.payload;
-    const response = await decryptWallet({password, wallet, walletVersion});
+    console.log('wallet', wallet)
+    const response = await decryptWallet({password, wallet, walletVersion: wallet?.version || walletVersion});
     event.source.postMessage(
       {
         requestId: request.requestId,