diff --git a/src/io.rs b/src/io.rs
index 877927b..77f4b4a 100644
--- a/src/io.rs
+++ b/src/io.rs
@@ -1,5 +1,3 @@
-// src/io.rs
-
 use std::{
     collections::HashMap,
     io::{Error, ErrorKind, Read, Result, Write},
diff --git a/src/lib.rs b/src/lib.rs
index 7925435..879b9f2 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -1,7 +1,6 @@
-// src/lib.rs
-
 use std::collections::HashMap;
 pub mod io;
+#[cfg(test)]
 mod test;
 pub use io::{Reader, Writer};
 
diff --git a/src/test.rs b/src/test.rs
index 2712316..e216afb 100644
--- a/src/test.rs
+++ b/src/test.rs
@@ -1,5 +1,3 @@
-// src/test.rs
-
 use crate::{Tag, read_nbt, write_nbt};
 
 #[test]