diff --git a/files/initrd/opt/rr/menu.sh b/files/initrd/opt/rr/menu.sh
index 7acefd45..eca59acf 100755
--- a/files/initrd/opt/rr/menu.sh
+++ b/files/initrd/opt/rr/menu.sh
@@ -1280,8 +1280,8 @@ function make() {
     while true; do
       SIZE=256 # initrd-dsm + zImage-dsm ≈ 210M
       SPACELEFT=$(df -m ${PART3_PATH} 2>/dev/null | awk 'NR==2 {print $4}')
-      ZIMAGESIZE=$(du -m ${ORI_ZIMAGE_FILE} 2>/dev/null | awk '{print $1}')
-      RDGZSIZE=$(du -m ${ORI_RDGZ_FILE} 2>/dev/null | awk '{print $1}')
+      ZIMAGESIZE=$(du -m ${MOD_ZIMAGE_FILE} 2>/dev/null | awk '{print $1}')
+      RDGZSIZE=$(du -m ${MOD_RDGZ_FILE} 2>/dev/null | awk '{print $1}')
       SPACEALL=$((${SPACELEFT:-0} + ${ZIMAGESIZE:-0} + ${RDGZSIZE:-0}))
       [ ${SPACEALL:-0} -ge ${SIZE} ] && break
       echo -e "$(TEXT "No disk space left, please clean the cache and try again!")" >"${LOG_FILE}"
diff --git a/localbuild.sh b/localbuild.sh
index 852f0617..ddaadefd 100755
--- a/localbuild.sh
+++ b/localbuild.sh
@@ -189,11 +189,11 @@ function pack() {
     exit 1
   )
   sudo mount ${LOOPX}p2 "/tmp/mnt/p2" || (
-    echo -e "Can't mount ${LOOPX}p1."
+    echo -e "Can't mount ${LOOPX}p2."
     exit 1
   )
   sudo mount ${LOOPX}p3 "/tmp/mnt/p3" || (
-    echo -e "Can't mount ${LOOPX}p1."
+    echo -e "Can't mount ${LOOPX}p3."
     exit 1
   )
 
@@ -204,11 +204,11 @@ function pack() {
     exit 1
   )
   sudo cp -rf "${CHROOT_PATH}/mnt/p2/"* "/tmp/mnt/p2" || (
-    echo -e "Can't cp ${LOOPX}p1."
+    echo -e "Can't cp ${LOOPX}p2."
     exit 1
   )
   sudo cp -rf "${CHROOT_PATH}/mnt/p3/"* "/tmp/mnt/p3" || (
-    echo -e "Can't cp ${LOOPX}p1."
+    echo -e "Can't cp ${LOOPX}p3."
     exit 1
   )
   sudo sync