diff --git a/ostp-core/src/crypto/obfuscation.rs b/ostp-core/src/crypto/obfuscation.rs
index 4af4e89..a3dd059 100644
--- a/ostp-core/src/crypto/obfuscation.rs
+++ b/ostp-core/src/crypto/obfuscation.rs
@@ -198,3 +198,7 @@ pub fn deobfuscate_packet_inplace(raw: &mut [u8], key: &[u8; 8], is_handshake: b
         }
     }
 }
+
+#[cfg(test)]
+#[path = "obfuscation_tests.rs"]
+mod obfuscation_tests;
diff --git a/ostp-core/src/crypto/obfuscation_tests.rs b/ostp-core/src/crypto/obfuscation_tests.rs
index 47c871b..8942af9 100644
--- a/ostp-core/src/crypto/obfuscation_tests.rs
+++ b/ostp-core/src/crypto/obfuscation_tests.rs
@@ -1,6 +1,6 @@
 #[cfg(test)]
 mod tests {
-    use super::*;
+    use crate::crypto::obfuscation::*;
 
     /// Verifies that derive_all_secrets is deterministic — same input always
     /// produces the same output.