From 0a72e3cfe08289ef1962330546aa1532a9c88247 Mon Sep 17 00:00:00 2001
From: Ryo Onodera <ryoqun@gmail.com>
Date: Sat, 18 Jul 2020 00:08:44 +0900
Subject: [PATCH] Fix vote_balance_and_staked

---
 runtime/src/stakes.rs | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

diff --git a/runtime/src/stakes.rs b/runtime/src/stakes.rs
index cb68d79ce4..75cddb49f5 100644
--- a/runtime/src/stakes.rs
+++ b/runtime/src/stakes.rs
@@ -88,10 +88,15 @@ impl Stakes {
     }
 
     pub fn vote_balance_and_staked(&self) -> u64 {
-        self.vote_accounts
+        self.stake_delegations
             .iter()
-            .map(|(_pubkey, (staked, account))| staked + account.lamports)
-            .sum()
+            .map(|(_, stake_delegation)| stake_delegation.stake)
+            .sum::<u64>()
+            + self
+                .vote_accounts
+                .iter()
+                .map(|(_pubkey, (_staked, vote_account))| vote_account.lamports)
+                .sum::<u64>()
     }
 
     pub fn is_stake(account: &Account) -> bool {