Fix custom address handling in chain rewards

skalenetwork · Nov 21, 2024 · 435d99e · 435d99e
1 parent 2495bc1
commit 435d99e
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/components/delegation/ChainRewards.tsx b/src/components/delegation/ChainRewards.tsx
@@ -94,6 +94,8 @@ const ChainRewards: React.FC<ChainRewardsProps> = ({
   const { data: walletClient } = useWagmiWalletClient()
   const { switchChainAsync } = useWagmiSwitchNetwork()
 
+  const addr = customAddress ?? address
+
   useEffect(() => {
     loadData()
     const intervalId = setInterval(loadData, DEFAULT_UPDATE_INTERVAL_MS)
@@ -121,7 +123,7 @@ const ChainRewards: React.FC<ChainRewardsProps> = ({
       setTokenUrl(getExplorerUrlForAddress(network, paymasterChain, tokenAddress))
       setSklToken(skl)
     }
-    setTokenBalance(await skl.balanceOf(address))
+    setTokenBalance(await skl.balanceOf(addr))
   }
 
   async function retrieveRewards() {