fix: staking rewards apy KeyError (#590)

yearn · Sep 5, 2023 · 555fbaf · 555fbaf
1 parent df8ccfd
commit 555fbaf
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/yearn/apy/staking_rewards.py b/yearn/apy/staking_rewards.py
@@ -22,7 +22,8 @@ async def get_staking_rewards_apr(vault, samples: ApySamples):
     if await staking_pool.periodFinish.coroutine() < now:
         return 0
 
-    rewards_vault = vault.registry._vaults[await staking_pool.rewardsToken.coroutine()]
+    vaults = await vault.registry.vaults
+    rewards_vault = vaults[await staking_pool.rewardsToken.coroutine()]
     (
         reward_rate, 
         rewards_vault_scale,