diff --git a/src/tests.rs b/src/tests.rs
index 7f6bb95..38c26c6 100644
--- a/src/tests.rs
+++ b/src/tests.rs
@@ -4,6 +4,7 @@
 mod tests {
     use tempfile::tempdir;
     use std::io::Write;
+    use std::str::FromStr;
     
     // Import the modules we need for testing
     use crate::compression_engine::{self, CompressionType};