diff --git a/afl/src/engines/mod.rs b/afl/src/engines/mod.rs
index ee35af800d..d8fec2ed31 100644
--- a/afl/src/engines/mod.rs
+++ b/afl/src/engines/mod.rs
@@ -380,6 +380,8 @@ where
     }
 }
 
+// TODO: no_std test
+#[cfg(feature = "std")]
 #[cfg(test)]
 mod tests {
 
@@ -403,7 +405,6 @@ mod tests {
         ExitKind::Ok
     }
 
-    #[cfg(feature = "std")]
     #[test]
     fn test_engine() {
         let mut rand = StdRand::new(0);
diff --git a/afl/src/stages/mutational.rs b/afl/src/stages/mutational.rs
index 154d3beeda..c95d4cd18e 100644
--- a/afl/src/stages/mutational.rs
+++ b/afl/src/stages/mutational.rs
@@ -2,6 +2,7 @@ use alloc::rc::Rc;
 use core::cell::RefCell;
 use core::marker::PhantomData;
 
+use crate::{fire_event, AflError};
 use crate::corpus::testcase::Testcase;
 use crate::engines::State;
 use crate::events::{EventManager, NewTestcaseEvent};
@@ -11,7 +12,6 @@ use crate::mutators::Mutator;
 use crate::stages::Corpus;
 use crate::stages::Stage;
 use crate::utils::Rand;
-use crate::{fire_event, AflError};
 
 // TODO multi mutators stage