diff --git a/programs/steward/src/state/steward_state.rs b/programs/steward/src/state/steward_state.rs
index bead7169..9ab234ac 100644
--- a/programs/steward/src/state/steward_state.rs
+++ b/programs/steward/src/state/steward_state.rs
@@ -546,12 +546,9 @@ impl StewardState {
         self.delegations[num_pool_validators] = Delegation::default();
         self.instant_unstake.set(num_pool_validators, false)?;
         self.progress.set(num_pool_validators, false)?;
-
-        if marked_for_regular_removal {
-            self.validators_to_remove.set(index, false)?;
-        } else {
-            self.validators_for_immediate_removal.set(index, false)?;
-        }
+        self.validators_to_remove.set(num_pool_validators, false)?;
+        self.validators_for_immediate_removal
+            .set(num_pool_validators, false)?;
 
         Ok(())
     }