fix(staking): census entries should instead iterate on StakeKey<Address>

witnet · Dec 9, 2024 · 06dae81 · 06dae81
1 parent ae52341
commit 06dae81
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/data_structures/src/staking/stakes.rs b/data_structures/src/staking/stakes.rs
@@ -198,8 +198,8 @@ where
         capability: Capability,
         epoch: Epoch,
         strategy: CensusStrategy,
-    ) -> Box<dyn Iterator<Item = Address> + '_> {
-        let iterator = self.rank(capability, epoch).map(|(address, _)| address);
+    ) -> Box<dyn Iterator<Item = StakeKey<Address>> + '_> {
+        let iterator = self.by_rank(capability, epoch).map(|(address, _)| address);
 
         match strategy {
             CensusStrategy::All => Box::new(iterator),