From 1be7ee51be8e471c7cb1754764939e29f5bcf99c Mon Sep 17 00:00:00 2001
From: Pankaj Garg <pankaj@solana.com>
Date: Tue, 19 Mar 2019 11:49:06 -0700
Subject: [PATCH] Fix potential crash in banking stage

---
 core/src/banking_stage.rs | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/core/src/banking_stage.rs b/core/src/banking_stage.rs
index 71f2d69b2..2b9a55df7 100644
--- a/core/src/banking_stage.rs
+++ b/core/src/banking_stage.rs
@@ -150,7 +150,9 @@ impl BankingStage {
         // Buffer the packets if I am the next leader
         // or, if it was getting sent to me
         let leader_id = match poh_recorder.lock().unwrap().bank() {
-            Some(bank) => leader_schedule_utils::slot_leader_at(bank.slot() + 1, &bank).unwrap(),
+            Some(bank) => {
+                leader_schedule_utils::slot_leader_at(bank.slot() + 1, &bank).unwrap_or_default()
+            }
             None => rcluster_info
                 .leader_data()
                 .map(|x| x.id)