safe arith in upgrade to electra

ethDreamer · Apr 25, 2024 · 9ee134d · 9ee134d
1 parent bd9b179
commit 9ee134d
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/consensus/state_processing/src/upgrade/electra.rs b/consensus/state_processing/src/upgrade/electra.rs
@@ -1,3 +1,4 @@
+use safe_arith::SafeArith;
 use std::mem;
 use types::{
     BeaconState, BeaconStateElectra, BeaconStateError as Error, ChainSpec, EpochCache, EthSpec,
@@ -19,7 +20,7 @@ pub fn upgrade_to_electra<E: EthSpec>(
         .map(|v| v.exit_epoch)
         .max()
         .unwrap_or(epoch)
-        + 1;
+        .safe_add(1)?;
 
     let consolidation_balance_to_consume = pre_state.get_consolidation_churn_limit(spec)?;
     let earliest_consolidation_epoch = spec.compute_activation_exit_epoch(epoch)?;