diff --git a/ci/build_docs.sh b/ci/build_docs.sh
index 4bc3e63f..38936c08 100755
--- a/ci/build_docs.sh
+++ b/ci/build_docs.sh
@@ -2,11 +2,11 @@
 
 set -euo pipefail
 
-export RAPIDS_VERSION_NUMBER="$(rapids-version-major-minor)"
-
 rapids-logger "Create test conda environment"
 . /opt/conda/etc/profile.d/conda.sh
 
+RAPIDS_VERSION="$(rapids-version)"
+
 rapids-dependency-file-generator \
     --output conda \
     --file-key docs \
@@ -24,7 +24,8 @@ PYTHON_CHANNEL=$(rapids-download-conda-from-s3 python)
 rapids-mamba-retry install \
     --channel "${CPP_CHANNEL}" \
     --channel "${PYTHON_CHANNEL}" \
-    cucim libcucim
+    "cucim=${RAPIDS_VERSION}" \
+    "libcucim=${RAPIDS_VERSION}"
 
 export RAPIDS_DOCS_DIR="$(mktemp -d)"
 
@@ -35,4 +36,4 @@ mkdir -p "${RAPIDS_DOCS_DIR}/cucim/"html
 mv _html/* "${RAPIDS_DOCS_DIR}/cucim/html"
 popd
 
-rapids-upload-docs
+RAPIDS_VERSION_NUMBER="$(rapids-version-major-minor)" rapids-upload-docs