diff --git a/main.py b/main.py
index 4266727..6a5a5a5 100644
--- a/main.py
+++ b/main.py
@@ -294,16 +294,16 @@ for currentRow in tqdm(range(new_height), "计算新像素点中"):
 newimgArray = [66, 77]
 
 if (angle in [0, 180]):
-    rowLength = floor(width * bpp / 8)
+    rowLength = floor(new_width * bpp / 8)
     while (rowLength % 4 != 0 or rowLength == 0):
         rowLength += 1
     new_fileSize = 54 + rowLength * new_height  #文件头中的文件大小
     new_imageSize = rowLength * new_height
 else:
-    rowLength = floor(height * bpp / 8)
+    rowLength = floor(new_height * bpp / 8)
     while (rowLength % 4 != 0 or rowLength == 0):
         rowLength += 1
-    new_fileSize = 54 + rowLength * new_height  #文件头中的文件大小
+    new_fileSize = 54 + rowLength * new_width  #文件头中的文件大小
     new_imageSize = rowLength * new_width
 
 # 新文件头