From 35e219e1f63ae55eee860fbc1770bbc037d45a64 Mon Sep 17 00:00:00 2001
From: Matan Kushner <hello@matchai.me>
Date: Fri, 12 Apr 2019 19:13:14 -0400
Subject: [PATCH] Fix up benchmarks

---
 benches/benchmarks.rs | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/benches/benchmarks.rs b/benches/benchmarks.rs
index 5da21f04c..4bd3d31b4 100644
--- a/benches/benchmarks.rs
+++ b/benches/benchmarks.rs
@@ -5,7 +5,7 @@ extern crate test;
 #[cfg(test)]
 mod tests {
     use clap::{App, Arg};
-    use starship::{modules, print};
+    use starship::modules;
     use test::Bencher;
 
     // #[bench]
@@ -27,7 +27,7 @@ mod tests {
                 .get_matches_from(vec!["starship", "0"]);
 
             let segment = modules::handle("char", &args);
-            print::stringify_segment(segment)
+            segment.output()
         });
     }
 
@@ -39,7 +39,7 @@ mod tests {
                 .get_matches_from(vec!["starship", "0"]);
 
             let segment = modules::handle("dir", &args);
-            print::stringify_segment(segment)
+            segment.output()
         });
     }
 
@@ -51,7 +51,7 @@ mod tests {
                 .get_matches_from(vec!["starship", "0"]);
 
             let segment = modules::handle("line_break", &args);
-            print::stringify_segment(segment)
+            segment.output()
         });
     }
 
@@ -63,7 +63,7 @@ mod tests {
                 .get_matches_from(vec!["starship", "0"]);
 
             let segment = modules::handle("nodejs", &args);
-            print::stringify_segment(segment)
+            segment.output()
         });
     }
 }