load balances on removal

0xsequence · Nov 5, 2024 · 4a3c53d · 4a3c53d
1 parent d62e32b
commit 4a3c53d
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/src/stores/CollectibleStore.ts b/src/stores/CollectibleStore.ts
@@ -194,5 +194,9 @@ export class CollectibleStore {
         c.tokenId !== collectibleInfo.collectibleInfoParams.tokenId
     )
     this.local.userCollectibles.set(filtered)
+
+    this.isFetchingCollectibleInfo.set(true)
+    this.loadBalances()
+    this.isFetchingCollectibleInfo.set(false)
   }
 }
diff --git a/src/stores/TokenStore.ts b/src/stores/TokenStore.ts
@@ -231,6 +231,14 @@ export class TokenStore {
       .filter(b => !(b.chainId === token.chainId && b.contractAddress === token.address))
 
     this.balances.set(filteredBalances)
+
+    const accountAddress = this.store.get(AuthStore).accountAddress.get()
+
+    if (accountAddress) {
+      this.isFetchingBalances.set(true)
+      await this.loadUserAddedTokenBalance(accountAddress, token)
+      this.isFetchingBalances.set(false)
+    }
   }
 
   async getTokenInfo(chainId: number, address: string): Promise<UserAddedTokenInitialInfo> {