diff --git a/eth2/types/src/attestation_duty.rs b/eth2/types/src/attestation_duty.rs
index 80d912a83..9612dd868 100644
--- a/eth2/types/src/attestation_duty.rs
+++ b/eth2/types/src/attestation_duty.rs
@@ -6,4 +6,5 @@ pub struct AttestationDuty {
     pub slot: Slot,
     pub shard: Shard,
     pub committee_index: usize,
+    pub comittee_size: usize,
 }
diff --git a/validator_client/src/duties/epoch_duties.rs b/validator_client/src/duties/epoch_duties.rs
index 0a4f73f72..d0753a2a6 100644
--- a/validator_client/src/duties/epoch_duties.rs
+++ b/validator_client/src/duties/epoch_duties.rs
@@ -23,9 +23,7 @@ pub struct WorkInfo {
 #[derive(Debug, PartialEq, Clone, Copy, Default)]
 pub struct EpochDuty {
     pub block_production_slot: Option<Slot>,
-    pub attestation_slot: Slot,
-    pub attestation_shard: u64,
-    pub committee_index: u64,
+    pub attestation_duty: AttestationDuty,
 }
 
 impl EpochDuty {
@@ -39,12 +37,8 @@ impl EpochDuty {
 
         // if the validator is required to attest to a shard, create the data
         let mut attestation_duty = None;
-        if self.attestation_slot == slot {
-            attestation_duty = Some(AttestationDuty {
-                slot,
-                shard: self.attestation_shard,
-                committee_index: self.committee_index as usize,
-            });
+        if self.attestation_duty.slot == slot {
+            attestation_duty = self.attestation_duty
         }
 
         if produce_block | attestation_duty.is_some() {
diff --git a/validator_client/src/duties/grpc.rs b/validator_client/src/duties/grpc.rs
index 66bf368a2..f0d892e98 100644
--- a/validator_client/src/duties/grpc.rs
+++ b/validator_client/src/duties/grpc.rs
@@ -47,11 +47,17 @@ impl BeaconNodeDuties for ValidatorServiceClient {
                     None
                 }
             };
+
+            let attestation_duty = AttestationDuty {
+            slot: Slot::from(active_duty.get_attestation_slot()),
+            shard: active_duty.get_attestation_shard(),
+            committee_index: active_duty.get_committee_index(),
+            comittee_size: 10,
+            }
+
             let epoch_duty = EpochDuty {
                 block_production_slot,
-                attestation_slot: Slot::from(active_duty.get_attestation_slot()),
-                attestation_shard: active_duty.get_attestation_shard(),
-                committee_index: active_duty.get_committee_index(),
+                attestation_duty,
             };
             epoch_duties.insert(pub_keys[index].clone(), Some(epoch_duty));
         }
diff --git a/validator_client/src/duties/mod.rs b/validator_client/src/duties/mod.rs
index 9f0998567..d52cc3254 100644
--- a/validator_client/src/duties/mod.rs
+++ b/validator_client/src/duties/mod.rs
@@ -30,7 +30,6 @@ pub enum UpdateOutcome {
 #[derive(Debug, PartialEq)]
 pub enum Error {
     DutiesMapPoisoned,
-    EpochMapPoisoned,
     BeaconNodeDutiesError(BeaconNodeDutiesError),
     UnknownEpoch,
     UnknownValidator,