From 3c02374b396f52f1e05a8b33190752d024026f22 Mon Sep 17 00:00:00 2001
From: jeremy-babylonchain <jeremy.tsang@babylonchain.io>
Date: Fri, 13 Sep 2024 17:06:55 +0800
Subject: [PATCH] resolve comments

---
 src/app/components/Delegations/Delegations.tsx    |  1 +
 src/app/context/api/DelegationsPointsProvider.tsx | 11 +++++++++--
 2 files changed, 10 insertions(+), 2 deletions(-)

diff --git a/src/app/components/Delegations/Delegations.tsx b/src/app/components/Delegations/Delegations.tsx
index 591a31f6..2a47465e 100644
--- a/src/app/components/Delegations/Delegations.tsx
+++ b/src/app/components/Delegations/Delegations.tsx
@@ -62,6 +62,7 @@ export const Delegations: React.FC<DelegationsProps> = ({
       publicKeyNoCoord={publicKeyNoCoord}
       delegationsAPI={delegationsAPI}
       isWalletConnected={isWalletConnected}
+      address={address}
     >
       <DelegationsContent
         delegationsAPI={delegationsAPI}
diff --git a/src/app/context/api/DelegationsPointsProvider.tsx b/src/app/context/api/DelegationsPointsProvider.tsx
index 3560b730..d90d50ab 100644
--- a/src/app/context/api/DelegationsPointsProvider.tsx
+++ b/src/app/context/api/DelegationsPointsProvider.tsx
@@ -30,11 +30,18 @@ interface DelegationsPointsProviderProps {
   publicKeyNoCoord: string;
   delegationsAPI: Delegation[];
   isWalletConnected: boolean;
+  address: string;
 }
 
 export const DelegationsPointsProvider: React.FC<
   DelegationsPointsProviderProps
-> = ({ children, publicKeyNoCoord, delegationsAPI, isWalletConnected }) => {
+> = ({
+  children,
+  publicKeyNoCoord,
+  delegationsAPI,
+  isWalletConnected,
+  address,
+}) => {
   const [delegationPoints, setDelegationPoints] = useState<Map<string, number>>(
     new Map(),
   );
@@ -59,7 +66,7 @@ export const DelegationsPointsProvider: React.FC<
   };
 
   const { data, isLoading, error } = useQuery({
-    queryKey: ["delegationPoints", publicKeyNoCoord, delegationsAPI],
+    queryKey: ["delegationPoints", address, publicKeyNoCoord],
     queryFn: fetchAllPoints,
     enabled:
       isWalletConnected &&