diff --git a/libafl/src/schedulers/weighted.rs b/libafl/src/schedulers/weighted.rs
index 98d2246eeb..34d4b5f011 100644
--- a/libafl/src/schedulers/weighted.rs
+++ b/libafl/src/schedulers/weighted.rs
@@ -8,6 +8,8 @@ use hashbrown::HashMap;
 use libafl_bolts::rands::Rand;
 use serde::{Deserialize, Serialize};
 
+#[cfg(doc)]
+use crate::corpus::Testcase;
 use crate::{
     corpus::{Corpus, CorpusId, HasTestcase, SchedulerTestcaseMetadata},
     inputs::UsesInput,