diff --git a/src/main.rs b/src/main.rs
index beb91fd..123728c 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -163,7 +163,8 @@ fn main() -> Result<(), Box<dyn Error>> {
 
                 let mask = session.predict(&img)?;
                 let result_rgba = apply_mask(&img, &mask)?;
-                let output_path = output_directory.join(path.file_name().unwrap());
+                let mut output_path = output_directory.join(path.file_name().unwrap());
+                output_path.set_extension("png");
                 result_rgba.save(&output_path)?;
                 processed += 1;
                 info!(output = %output_path.display(), "Processed image saved");