diff --git a/build.sh b/build.sh
index a8247ab..b2571a4 100755
--- a/build.sh
+++ b/build.sh
@@ -108,11 +108,11 @@ run_stage(){
 				run_sub_stage
 			fi
 		done
+		PREV_ROOTFS_DIR="${ROOTFS_DIR}"
+		PREV_STAGE="${STAGE}"
+		PREV_STAGE_DIR="${STAGE_DIR}"
 	fi
 	unmount "${WORK_DIR}/${STAGE}"
-	PREV_STAGE="${STAGE}"
-	PREV_STAGE_DIR="${STAGE_DIR}"
-	PREV_ROOTFS_DIR="${ROOTFS_DIR}"
 	popd > /dev/null
 	log "End ${STAGE_DIR}"
 }