diff --git a/zspell/build.rs b/zspell/build.rs
index 13ace57..0445c1d 100644
--- a/zspell/build.rs
+++ b/zspell/build.rs
@@ -10,7 +10,7 @@ fn main() {
 
 const TEST_PREFIX: &str = "// autogenerated file, do not edit manually
 //! Runner for the `.test` integration test format
-#[rustfmt_skip] // FIXME: generate these in `OUTPUT_DIR` rather than `src`
+#![cfg_attr(rustfmt, rustfmt_skip)] // FIXME: generate these in `OUTPUT_DIR` rather than `src`
 
 use test_util::TestManager;
 ";
diff --git a/zspell/tests/auto_suite.rs b/zspell/tests/auto_suite.rs
index c8cf600..385c687 100644
--- a/zspell/tests/auto_suite.rs
+++ b/zspell/tests/auto_suite.rs
@@ -1,6 +1,6 @@
 // autogenerated file, do not edit manually
 //! Runner for the `.test` integration test format
-#[rustfmt_skip] // FIXME: generate these in `OUTPUT_DIR` rather than `src`
+#![cfg_attr(rustfmt, rustfmt_skip)] // FIXME: generate these in `OUTPUT_DIR` rather than `src`
 
 use test_util::TestManager;