From 0147bcf365db55110805fee41ef6cd0a647dd20c Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Cl=C3=A9ment=20Pr=C3=A9vost?=
 <998369+prevostc@users.noreply.github.com>
Date: Sat, 3 Aug 2024 13:55:30 +0200
Subject: [PATCH] Sometimes the boost is not created right away

---
 src/vault-interaction.ts | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/src/vault-interaction.ts b/src/vault-interaction.ts
index ed8ff7c..953d7c4 100644
--- a/src/vault-interaction.ts
+++ b/src/vault-interaction.ts
@@ -132,7 +132,12 @@ function updateInvestorVaultData(vault: BeefyVault, investor: Investor): Investo
     for (let i = 0; i < boosts.length; i++) {
       const boostAddress = boosts[i]
       const boostContract = BeefyBoostContract.bind(boostAddress)
-      rawSharesBalance = rawSharesBalance.plus(boostContract.balanceOf(investorAddress))
+      const boostBalanceRes = boostContract.try_balanceOf(investorAddress)
+      if (!boostBalanceRes.reverted) {
+        rawSharesBalance = rawSharesBalance.plus(boostBalanceRes.value)
+      } else {
+        log.warning("updateInvestorVaultData: boost balanceOf reverted {}", [boostAddress.toHexString()])
+      }
     }
   } else {
     log.error("updateInvestorVaultData: vault config not found {}", [vault.id.toHexString()])