Fix "stats by dao" calculations (solana-labs#1385)

jito-foundation · Feb 6, 2023 · b781a01 · b781a01
1 parent 48ff717
commit b781a01
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/hub/components/GlobalStats/data/index.ts b/hub/components/GlobalStats/data/index.ts
@@ -149,13 +149,19 @@ export async function fetchData(
     for (const governanceAddress of governanceAddrs) {
       // Check governance owned token accounts
       try {
+        const solWalletPk = await getNativeTreasuryAddress(
+          realm.programId,
+          governanceAddress,
+        );
         const tokenAccounts = await getTokenAmount(
           connection,
           governanceAddress,
         );
-        for (const tokenAccount of tokenAccounts.filter(
-          (ta) => !ta.account.amount.isZero(),
-        )) {
+        const moreTokenAccounts = await getTokenAmount(connection, solWalletPk);
+
+        for (const tokenAccount of tokenAccounts
+          .concat(moreTokenAccounts)
+          .filter((ta) => !ta.account.amount.isZero())) {
           updateTokenAmount(
             tokenAccount.account.mint,
             tokenAccount.account.amount,