diff --git a/base.dockerfile b/base.dockerfile
index 54176cd..6b463f4 100644
--- a/base.dockerfile
+++ b/base.dockerfile
@@ -82,7 +82,7 @@ RUN git clone -b ${ROCBLAS_BRANCH} https://github.com/ROCmSoftwarePlatform/rocBL
 WORKDIR /root/rocBLAS
 RUN rm -rf library/src/blas3/Tensile/Logic/asm_full/r9nano/
 RUN sed -i -e 's/rmake.py --install/rmake.py -a="${ROCBLAS_ARCH}" --install/g' install.sh
-RUN ./install.sh -d --cmake_install
+RUN ./install.sh -a ${ROCBLAS_ARCH} -d --cmake_install
 WORKDIR /root/rocBLAS/build/release
 COPY fix-rocblas.sh .
 RUN <<EOF