1 files changed, 3 insertions, 0 deletions
diff --git a/src/generator.rs b/src/generator.rs
index 6d18055..7a21f07 100644
--- a/src/generator.rs
+++ b/src/generator.rs
@@ -187,5 +187,8 @@ mod tests {
         let generator = MorseGenerator::new(12000, 600.0, 20.0);
         let result = generator.generate_wav_file("SOS", "test_sos.wav");
         assert!(result.is_ok());
+        
+        // Clean up test file
+        std::fs::remove_file("test_sos.wav").ok();
     }
 }