diff --git a/lib/codegen/src/lib.rs b/lib/codegen/src/lib.rs
index dce082e02a..0c7eb52989 100644
--- a/lib/codegen/src/lib.rs
+++ b/lib/codegen/src/lib.rs
@@ -48,10 +48,7 @@ extern crate alloc;
 extern crate failure;
 #[macro_use]
 extern crate failure_derive;
-#[cfg(test)]
-#[macro_use]
-extern crate target_lexicon;
-#[cfg(not(test))]
+#[cfg_attr(test, macro_use)]
 extern crate target_lexicon;
 
 pub use context::Context;