diff --git a/src/lib.rs b/src/lib.rs
index 50c63e892b..ecc1cc9980 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -1,7 +1,10 @@
-#![cfg_attr(not(any(test, feature = "std")), no_std)]
+#![no_std]
 //! A super simple initfs, only meant to be loaded into RAM by the bootloader, and then directly be
 //! read.
 
+#[cfg(any(test, feature = "std"))]
+extern crate std;
+
 use core::convert::{TryFrom, TryInto};
 
 pub mod types;