diff --git a/src/tap_test.rs b/src/tap_test.rs
index ef9e534..1e22db7 100644
--- a/src/tap_test.rs
+++ b/src/tap_test.rs
@@ -7,7 +7,11 @@ use alloc::{
     vec::Vec,
 };
 use core::fmt::Write;
+
+#[cfg(feature = "std")]
 use std::fmt;
+#[cfg(not(feature = "std"))]
+use core::fmt;
 
 use crate::{NOT_OK_SYMBOL, OK_SYMBOL};