diff --git a/pallets/subtensor/src/coinbase.rs b/pallets/subtensor/src/coinbase.rs index ce89fb407..017a1e12e 100644 --- a/pallets/subtensor/src/coinbase.rs +++ b/pallets/subtensor/src/coinbase.rs @@ -85,7 +85,7 @@ impl Pallet { ); // --- 4.3 Set last step counter. - Self::set_blocks_since_last_step(*netuid, 0); + Self::set_blocks_since_last_step(*netuid, current_block); Self::set_last_mechanism_step_block(*netuid, current_block); // --- 4.4 Distribute owner take. diff --git a/pallets/subtensor/tests/block_step.rs b/pallets/subtensor/tests/block_step.rs index c3318b181..dc506ec4b 100644 --- a/pallets/subtensor/tests/block_step.rs +++ b/pallets/subtensor/tests/block_step.rs @@ -1765,16 +1765,16 @@ fn test_epoch_runs_when_registration_disabled() { step_block(1); // Now block 2 // Verify blocks since last step was set - assert_eq!(SubtensorModule::get_blocks_since_last_step(netuid_off), 1); + assert_eq!(SubtensorModule::get_blocks_since_last_step(netuid_off), 0); // Step to the next epoch block let epoch_block: u16 = tempo; - step_block(epoch_block); + step_block(epoch_block); // Now block 3 // Verify blocks since last step was set, this indicates we ran the epoch assert_eq!( SubtensorModule::get_blocks_since_last_step(netuid_off), - 0_u64 + 3_u64 ); assert!(SubtensorModule::get_loaded_emission_tuples(netuid_off).is_some()); });