diff --git a/.github/workflows/pre-check.yml b/.github/workflows/pre-check.yml
index 0b8d414..f8f75d0 100644
--- a/.github/workflows/pre-check.yml
+++ b/.github/workflows/pre-check.yml
@@ -21,7 +21,7 @@ jobs:
           docker buildx build \
             -f docker/Dockerfile.cuda13.x.ubuntu22.04 \
             --platform linux/amd64 \
-            --target package \
+            --target build \
             --output type=local,dest=dist \
             .
       - name: List artifacts
diff --git a/docker/Dockerfile.cuda13.x.ubuntu22.04 b/docker/Dockerfile.cuda13.x.ubuntu22.04
index 8e6a453..1ca8a1c 100644
--- a/docker/Dockerfile.cuda13.x.ubuntu22.04
+++ b/docker/Dockerfile.cuda13.x.ubuntu22.04
@@ -24,7 +24,7 @@ ARG BUILD_DATE=20251221
 ###########################
 # Build Stage
 ###########################
-FROM nvidia/cuda:${CUDA_VERSION}-devel-ubuntu${UBUNTU_VERSION}
+FROM nvidia/cuda:${CUDA_VERSION}-devel-ubuntu${UBUNTU_VERSION} AS build
 
 # Re-declare build args for this stage (values are inherited)
 ARG UBUNTU_VERSION