11import { CurrencyBalance } from '@centrifuge/centrifuge-js'
2- import { useEvmProvider } from '@centrifuge/centrifuge-react'
2+ import { useWallet } from '@centrifuge/centrifuge-react'
33import { ethers } from 'ethers'
44import { useQuery } from 'react-query'
55import { isTestEnv } from '../../../src/config'
@@ -21,9 +21,10 @@ export const cfgConfig = isTestEnv
2121 }
2222
2323export const useTokenBalance = ( userAddress : string | undefined ) => {
24- const provider = useEvmProvider ( )
24+ const { evm } = useWallet ( )
25+ const provider = evm . getProvider ( evm . chainId ! )
2526 return useQuery (
26- [ 'tokenBalance' , userAddress ] ,
27+ [ 'tokenBalance' , userAddress , evm . chainId ] ,
2728 async ( ) => {
2829 const tokens = await Promise . allSettled ( [
2930 new ethers . Contract ( cfgConfig . legacy , ABI , provider ) . balanceOf ( userAddress ! ) ,
@@ -54,9 +55,10 @@ export const useTokenBalance = (userAddress: string | undefined) => {
5455}
5556
5657export const useCheckAllowance = ( userAddress : string | undefined ) => {
57- const provider = useEvmProvider ( )
58+ const { evm } = useWallet ( )
59+ const provider = evm . getProvider ( evm . chainId ! )
5860 return useQuery (
59- [ 'checkAllowance' , userAddress ] ,
61+ [ 'checkAllowance' , userAddress , evm . chainId ] ,
6062 async ( ) => {
6163 const allowance = await new ethers . Contract ( cfgConfig . legacy , IOU_ABI , provider ) . allowance (
6264 userAddress ! ,
0 commit comments