diff --git a/docker/rocm/Dockerfile b/docker/rocm/Dockerfile
index 65091664f..d13bcaead 100644
--- a/docker/rocm/Dockerfile
+++ b/docker/rocm/Dockerfile
@@ -75,6 +75,7 @@ COPY docker/rocm/rootfs/ /
 FROM scratch AS rocm-dist
 
 ARG ROCM
+ARG AMDGPU
 
 COPY --from=rocm /opt/rocm-$ROCM/bin/rocminfo /opt/rocm-$ROCM/bin/migraphx-driver /opt/rocm-$ROCM/bin/
 COPY --from=rocm /opt/rocm-$ROCM/share/miopen/db/*$AMDGPU* /opt/rocm-$ROCM/share/miopen/db/