diff --git a/app/MindWork AI Studio/Tools/FileInfoExtensions.cs b/app/MindWork AI Studio/Tools/FileInfoExtensions.cs
index 8d3d81d..c300262 100644
--- a/app/MindWork AI Studio/Tools/FileInfoExtensions.cs	
+++ b/app/MindWork AI Studio/Tools/FileInfoExtensions.cs	
@@ -2,20 +2,16 @@ namespace AIStudio.Tools;
 
 public static class FileInfoExtensions
 {
+    /// <summary>
+    /// Returns the file size in human-readable format.
+    /// </summary>
+    /// <param name="fileInfo">The file info object.</param>
+    /// <returns>The file size in human-readable format.</returns>
     public static string FileSize(this FileInfo fileInfo)
     {
         if (!fileInfo.Exists)
             return "N/A";
 
-        var size = fileInfo.Length;
-        string[] sizes = { "B", "kB", "MB", "GB", "TB" };
-        var order = 0;
-        while (size >= 1024 && order < sizes.Length - 1)
-        {
-            order++;
-            size /= 1024;
-        }
-
-        return $"{size:0.##} {sizes[order]}";
+        return fileInfo.Length.FileSize();
     }
 }
\ No newline at end of file