diff --git a/bin/src/main.rs b/bin/src/main.rs
index d934be2..954da71 100644
--- a/bin/src/main.rs
+++ b/bin/src/main.rs
@@ -103,15 +103,18 @@ fn process_html(input: &str) -> Result<String, sanitize_html::errors::SanitizeEr
     //Ok(input.to_owned())
 }
 
-fn write_to_test_path(uid: &str, message: &[u8]) {
-    let test_path: PathBuf = [Path::new("tests/data"), Path::new(&format!("{}.eml", &uid))]
-        .iter()
-        .collect();
+fn write_to_test_path(msg: &Message) {
+    let test_path: PathBuf = [
+        Path::new("tests/data"),
+        Path::new(&format!("{}.eml", &msg.get_uid())),
+    ]
+    .iter()
+    .collect();
 
     let _ = OpenOptions::new()
         .write(true)
         .create(true)
         .open(test_path)
         .expect("Could not open file fir writing")
-        .write_all(&message);
+        .write_all(&msg.data);
 }