diff --git a/.github/scripts/docker/build_and_push.sh b/.github/scripts/docker/build_and_push.sh
index 17a5a36..f569ee3 100755
--- a/.github/scripts/docker/build_and_push.sh
+++ b/.github/scripts/docker/build_and_push.sh
@@ -2,8 +2,15 @@
 
 # arch=x86_64
 arch=aarch64
+
+if [ "$arch" = "x86_64" ]; then
+    platform=linux/amd64
+elif [ "$arch" = "aarch64" ]; then
+    platform=linux/arm64
+fi
+
 image_base=quay.io/pypa/manylinux_2_28_${arch}
 tag=samwelborn/stempy_wheel_builder_${arch}
 
-docker build . --platform=linux/amd64,linux/arm64 -t $tag --build-arg BASE_IMAGE=$image_base --build-arg ARCH=$arch
+docker build . --platform=$platform -t $tag --build-arg BASE_IMAGE=$image_base --build-arg ARCH=$arch
 docker push $tag