wip

Author: ibhatt-jumptrading

src/block.rs

@@ -558,8 +558,10 @@ pub fn execute_block(context: BlockContext) -> Option<BlockEffects> {
     let l_sched = leader_schedule_utils::leader_schedule(current_epoch, &bank).unwrap();
     let epoch_schedule = bank.epoch_schedule();
     let (_, slot_index) = epoch_schedule.get_epoch_and_slot_index(current_slot);
-    let leader = l_sched.get_vote_key_at_slot_index(slot_index as usize).unwrap();
-    bank.set_collector_id_for_tests(leader.clone());
+    let leader = l_sched.get_vote_key_at_slot_index(slot_index as usize);
+    if !leader.is_none() {
+        bank.set_collector_id_for_tests(leader.unwrap().clone());
+    }
 
     bank.distribute_partitioned_epoch_rewards();