PATCH: Carry over epoch maintenance flag (#64)

Co-authored-by: Christian Krueger <[email protected]>
jito-foundation · Jul 29, 2024 · ad711ce · ad711ce
1 parent f83602d
commit ad711ce
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/programs/steward/src/state/steward_state.rs b/programs/steward/src/state/steward_state.rs
@@ -449,7 +449,12 @@ impl StewardState {
         self.stake_deposit_unstake_total = 0;
         self.delegations = [Delegation::default(); MAX_VALIDATORS];
         self.instant_unstake = BitMask::default();
+
+        let has_epoch_maintenance = self.has_flag(EPOCH_MAINTENANCE);
         self.clear_flags();
+        if has_epoch_maintenance {
+            self.set_flag(EPOCH_MAINTENANCE);
+        }
 
         Ok(())
     }