diff --git a/build.sh b/build.sh
index b496d79..9f41838 100755
--- a/build.sh
+++ b/build.sh
@@ -8,7 +8,11 @@ OUTPUT_DIR=$(pwd)/_output
 IMAGE_TAG=afio_builder
 
 rm -rf "$OUTPUT_DIR"
+mkdir -p "$OUTPUT_DIR"
 echo "Build docker image: $IMAGE_TAG"
-docker buildx build -t $IMAGE_TAG .
+docker buildx build -t $IMAGE_TAG \
+    --iidfile "$OUTPUT_DIR"/iddfile \
+    .
+IMAGE_ID=$(cat "$OUTPUT_DIR"/iddfile)
 echo "Run docker ..."
-docker run --rm -t -v "$OUTPUT_DIR":/output $IMAGE_TAG
+docker run --rm -t -v "$OUTPUT_DIR":/output "$IMAGE_ID"