diff --git a/.github/workflows/publish-docker-rocm.yml b/.github/workflows/publish-docker-rocm.yml
index cce5891..b2b1792 100644
--- a/.github/workflows/publish-docker-rocm.yml
+++ b/.github/workflows/publish-docker-rocm.yml
@@ -47,4 +47,4 @@ jobs:
           file: Dockerfile
           tags: ghcr.io/${{ github.repository }}-rocm:${{ env.VERSION }}
           build-args: |
-            BUILD_TYPE=cuda
+            BUILD_TYPE=rocm