From ec35c1fc79cda28941bb4575aa21820c2c1bd31d Mon Sep 17 00:00:00 2001
From: Greg Fitzgerald <greg@solana.com>
Date: Tue, 26 Feb 2019 09:38:31 -0700
Subject: [PATCH] Fix leader scheduling in replay stage

---
 src/replay_stage.rs | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/src/replay_stage.rs b/src/replay_stage.rs
index 95117d5be4..edb39b2acb 100644
--- a/src/replay_stage.rs
+++ b/src/replay_stage.rs
@@ -89,11 +89,12 @@ impl ReplayStage {
         );
 
         let num_ticks = bank.tick_height();
-        let (mut num_ticks_to_next_vote, slot_height) = {
+        let (mut num_ticks_to_next_vote, slot_height, leader_id) = {
             let rl = leader_scheduler.read().unwrap();
             (
                 rl.num_ticks_left_in_slot(num_ticks),
                 rl.tick_height_to_slot(num_ticks),
+                rl.get_leader_for_slot(slot).expect("Leader not known"),
             )
         };
 
@@ -115,7 +116,7 @@ impl ReplayStage {
             entries_with_meta.push(EntryMeta {
                 tick_height: entry_tick_height,
                 slot,
-                slot_leader: bank.slot_leader(),
+                slot_leader: leader_id,
                 num_ticks_left_in_slot: num_ticks_to_next_vote,
                 parent_slot,
                 entry: entry.clone(),