Fix bug in balanceOf ERC-20

tuler · Oct 19, 2023 · 26efd8d · 26efd8d
1 parent 7c4fa53
commit 26efd8d
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/.changeset/fresh-zebras-worry.md b/.changeset/fresh-zebras-worry.md
@@ -0,0 +1,5 @@
+---
+"@deroll/wallet": patch
+---
+
+fix bug in balanceOf ERC-20
diff --git a/packages/wallet/src/wallet.ts b/packages/wallet/src/wallet.ts
@@ -48,8 +48,8 @@ export class WalletAppImpl implements WalletApp {
             }
 
             // erc-20 balance
-            const erc20: Record<Address, bigint> = this.wallets[address] ?? {};
-            return erc20[tokenOrAddress as Address] ?? 0n;
+            const wallet = this.wallets[address] ?? { ether: 0n, erc20: {} };
+            return wallet.erc20[tokenOrAddress as Address] ?? 0n;
         } else {
             // if is address, normalize it
             if (isAddress(tokenOrAddress)) {