diff --git a/src/lib.rs b/src/lib.rs
index 1b15781..63a7975 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -70,7 +70,10 @@ pub fn get_display_size(size: i64) -> String {
     if size < SIZE_PB {
         return format!("{:.*}GB", 2, (size as f64) / 1024. / 1024. / 1024.);
     }
-    return format!("{:.*}PB", 2, (size as f64) / 1024. / 1024. / 1024. / 1024.);
+    if size < SIZE_TB {
+        return format!("{:.*}PB", 2, (size as f64) / 1024. / 1024. / 1024. / 1024.);
+    }
+    return format!("{:.*}TB", 2, (size as f64) / 1024. / 1024. / 1024. / 1024. / 1024.);
 }
 
 pub fn run_command_and_wait(cmd: &mut Command) -> io::Result<()> {