1 files changed, 3 insertions, 4 deletions

diff --git a/scripts/boot-image.sh b/scripts/boot-image.sh
index e8c093c..f70dcd4 100755
--- a/scripts/boot-image.sh
+++ b/scripts/boot-image.sh
@@ -20,11 +20,10 @@
 set -euo pipefail
 
 SCRIPT_DIR="$(cd "$(dirname "$0")" && pwd)"
-PROJECT_DIR="$(dirname "$SCRIPT_DIR")"
-
+PROJECT_DIR="$(cd "$SCRIPT_DIR/.." && pwd)"
 cd "$PROJECT_DIR"
 
-IMAGE="rs-dos.img"
+IMAGE="${1:-rs-dos.img}"
 
 # ---- config ----
 OVMF="${OVMF:-/usr/share/edk2/x64/OVMF.4m.fd}"
@@ -40,7 +39,7 @@ parse_boot_args "$@"
 # ---- build image if missing ----
 if [ ! -f "$IMAGE" ]; then
         echo "Image not found, building..."
-        "$SCRIPT_DIR/build-image.sh"
+        "$SCRIPT_DIR/mksystem.sh" --chain --gpt "$IMAGE"
 fi
 
 # ---- boot ----