diff --git a/.github/workflows/pr-docker.yaml b/.github/workflows/pr-docker.yaml
index 32159fcda3..3fe09fa8fc 100644
--- a/.github/workflows/pr-docker.yaml
+++ b/.github/workflows/pr-docker.yaml
@@ -47,7 +47,7 @@ jobs:
     name: ${{ matrix.IMAGE_NAME }}
     with:
       build-args: |
-        AWS_OFI_NCCL_VERSION=${{ matrix.AWS_OFI_NCCL_VERSION }}
+        EFA_INSTALLER_VERSION=${{ matrix.EFA_INSTALLER_VERSION }}
         BASE_IMAGE=${{ matrix.BASE_IMAGE }}
         CUDA_VERSION=${{ matrix.CUDA_VERSION }}
         MOFED_VERSION=${{ matrix.MOFED_VERSION }}
diff --git a/.github/workflows/release-docker.yaml b/.github/workflows/release-docker.yaml
index b2bd97b829..5331f6a0d5 100644
--- a/.github/workflows/release-docker.yaml
+++ b/.github/workflows/release-docker.yaml
@@ -42,7 +42,7 @@ jobs:
     name: ${{ matrix.IMAGE_NAME }}
     with:
       build-args: |
-        AWS_OFI_NCCL_VERSION=${{ matrix.AWS_OFI_NCCL_VERSION }}
+        EFA_INSTALLER_VERSION=${{ matrix.EFA_INSTALLER_VERSION }}
         BASE_IMAGE=${{ matrix.BASE_IMAGE }}
         CUDA_VERSION=${{ matrix.CUDA_VERSION }}
         MOFED_VERSION=${{ matrix.MOFED_VERSION }}