diff --git a/fuzzers/baby_fuzzer_wasm/src/lib.rs b/fuzzers/baby_fuzzer_wasm/src/lib.rs
index 20f57a3072..082dea4ba1 100644
--- a/fuzzers/baby_fuzzer_wasm/src/lib.rs
+++ b/fuzzers/baby_fuzzer_wasm/src/lib.rs
@@ -11,7 +11,7 @@ use libafl::{
     mutators::{havoc_mutations, StdScheduledMutator},
     observers::StdMapObserver,
     schedulers::QueueScheduler,
-    stages::{ExecutionCountRestartHelperMetadata, StdMutationalStage},
+    stages::{StdRestartHelper, StdMutationalStage},
     state::{HasSolutions, StdState},
     Fuzzer, StdFuzzer,
 };
@@ -44,7 +44,7 @@ pub fn fuzz() {
     // No concurrency in WASM so these accesses are not racing.
     unsafe {
         RegistryBuilder::register::<MapFeedbackMetadata<u8>>();
-        RegistryBuilder::register::<ExecutionCountRestartHelperMetadata>();
+        RegistryBuilder::register::<StdRestartHelper>();
     }
 
     let mut signals = [0u8; 64];