From edd99fafb6c42212bda9bcaa8f77d11c15515e23 Mon Sep 17 00:00:00 2001
From: Grant Wuerker <gwuerker@gmail.com>
Date: Thu, 25 Jul 2019 15:08:18 +0200
Subject: [PATCH] Getting attestation slot via helper method

---
 beacon_node/beacon_chain/src/beacon_chain.rs | 2 +-
 beacon_node/beacon_chain/src/fork_choice.rs  | 4 ++--
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/beacon_node/beacon_chain/src/beacon_chain.rs b/beacon_node/beacon_chain/src/beacon_chain.rs
index 67d928127..8a9421a1b 100644
--- a/beacon_node/beacon_chain/src/beacon_chain.rs
+++ b/beacon_node/beacon_chain/src/beacon_chain.rs
@@ -10,7 +10,7 @@ use operation_pool::{OperationPool, PersistedOperationPool};
 use parking_lot::{RwLock, RwLockReadGuard};
 use slot_clock::SlotClock;
 use state_processing::per_block_processing::errors::{
-    AttestationValidationError, AttesterSlashingValidationError, DepositValidationError,
+    AttesterSlashingValidationError, DepositValidationError,
     ExitValidationError, ProposerSlashingValidationError, TransferValidationError,
 };
 use state_processing::{
diff --git a/beacon_node/beacon_chain/src/fork_choice.rs b/beacon_node/beacon_chain/src/fork_choice.rs
index 92b683590..0f98ac9ce 100644
--- a/beacon_node/beacon_chain/src/fork_choice.rs
+++ b/beacon_node/beacon_chain/src/fork_choice.rs
@@ -174,13 +174,13 @@ impl<T: BeaconChainTypes> ForkChoice<T> {
             &attestation.aggregation_bitfield,
         )?;
 
-        let target_slot = attestation.data.target_epoch.start_slot(T::EthSpec::slots_per_epoch());
+        let block_slot = state.get_attestation_slot(&attestation.data)?;
 
         Ok(validator_indices
             .iter()
             .find(|&&v| {
                 match self.backend.latest_message(v) {
-                    Some((_, slot)) => target_slot > slot,
+                    Some((_, slot)) => block_slot > slot,
                     None => true
                 }
             }).is_some())