Merge pull request #32 from darakian/use-candidate-name-for-results

Use get_candidate_name() for results output

1 files changed, 2 insertions, 2 deletions

diff --git a/src/main.rs b/src/main.rs
index 4be89ea..adb56b7 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -175,7 +175,7 @@ fn write_results_to_file(fmt: PrintFmt, shared_files: &Vec<&Fileinfo>, unique_fi
         PrintFmt::Standard => {
             output.write_fmt(format_args!("Duplicates:\n")).unwrap();
             for file in shared_files.into_iter(){
-                let title = file.get_paths().get(0).unwrap().file_name().unwrap().to_str().unwrap();
+                let title = file.get_candidate_name();
                 output.write_fmt(format_args!("{}\n", title)).unwrap();
                 for entry in file.get_paths().iter(){
                     output.write_fmt(format_args!("\t{}\n", entry.as_path().to_str().unwrap())).unwrap();
@@ -183,7 +183,7 @@ fn write_results_to_file(fmt: PrintFmt, shared_files: &Vec<&Fileinfo>, unique_fi
             }
             output.write_fmt(format_args!("Singletons:\n")).unwrap();
             for file in unique_files.into_iter(){
-                let title = file.get_paths().get(0).unwrap().file_name().unwrap().to_str().unwrap();
+                let title = file.get_candidate_name();
                 output.write_fmt(format_args!("{}\n", title)).unwrap();
                 for entry in file.get_paths().iter(){
                     output.write_fmt(format_args!("\t{}\n", entry.as_path().to_str().unwrap())).unwrap();