Improve logic

paritytech · Jan 10, 2025 · 4bc3223 · 4bc3223
1 parent 2359bcc
commit 4bc3223
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/substrate/frame/staking/src/mock.rs b/substrate/frame/staking/src/mock.rs
@@ -545,12 +545,8 @@ impl ExtBuilder {
 
 		if self.initialize_first_session {
 			ext.execute_with(|| {
-				System::run_to_block_with::<AllPalletsWithSystem>(
-					1,
-					frame_system::RunToBlockHooks::default().after_initialize(|_| {
-						Timestamp::set_timestamp(INIT_TIMESTAMP);
-					}),
-				);
+				System::run_to_block::<AllPalletsWithSystem>(1);
+				Timestamp::set_timestamp(INIT_TIMESTAMP);
 			});
 		}