From f0696af507d5fc83e542130e89528f19cf11bd8b Mon Sep 17 00:00:00 2001
From: Paul Hauner <paul@paulhauner.com>
Date: Fri, 25 Jun 2021 12:16:38 +1000
Subject: [PATCH] Remove old balance getter in ef_tests

---
 consensus/state_processing/src/per_epoch_processing/altair.rs | 1 -
 testing/ef_tests/src/cases/epoch_processing.rs                | 4 +++-
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/consensus/state_processing/src/per_epoch_processing/altair.rs b/consensus/state_processing/src/per_epoch_processing/altair.rs
index 988d79bfce7..dd93ccab216 100644
--- a/consensus/state_processing/src/per_epoch_processing/altair.rs
+++ b/consensus/state_processing/src/per_epoch_processing/altair.rs
@@ -45,7 +45,6 @@ pub fn process_epoch<T: EthSpec>(
     // Slashings.
     process_slashings(
         state,
-        // FIXME(paul): could this be invalidated by rewards/penalties?
         participation_cache.current_epoch_total_active_balance(),
         spec.proportional_slashing_multiplier_altair,
         spec,
diff --git a/testing/ef_tests/src/cases/epoch_processing.rs b/testing/ef_tests/src/cases/epoch_processing.rs
index 695042baa4e..42177e07f56 100644
--- a/testing/ef_tests/src/cases/epoch_processing.rs
+++ b/testing/ef_tests/src/cases/epoch_processing.rs
@@ -142,7 +142,9 @@ impl<E: EthSpec> EpochTransition<E> for Slashings {
             BeaconState::Altair(_) => {
                 process_slashings(
                     state,
-                    state.get_total_active_balance(spec)?,
+                    altair::ParticipationCache::new(state, spec)
+                        .unwrap()
+                        .current_epoch_total_active_balance(),
                     spec.proportional_slashing_multiplier_altair,
                     spec,
                 )?;