diff --git a/rusk-recovery/src/state/snapshot/stake.rs b/rusk-recovery/src/state/snapshot/stake.rs index bfad413ad..198422047 100644 --- a/rusk-recovery/src/state/snapshot/stake.rs +++ b/rusk-recovery/src/state/snapshot/stake.rs @@ -14,7 +14,8 @@ use super::wrapper::Wrapper; #[derive(Serialize, Deserialize, PartialEq, Eq)] pub struct GenesisStake { - pub(crate) address: Wrapper, + address: Wrapper, + owner: Option>, pub amount: Dusk, pub eligibility: Option, pub reward: Option, @@ -26,6 +27,7 @@ impl GenesisStake { } pub fn to_stake_keys(&self) -> StakeKeys { - StakeKeys::single_key(*self.address()) + let owner: &BlsPublicKey = self.owner.as_ref().unwrap_or(&self.address); + StakeKeys::new(*self.address(), *owner) } }