diff --git a/ci/release/update-version.sh b/ci/release/update-version.sh
index 0363b1f..113dcc6 100755
--- a/ci/release/update-version.sh
+++ b/ci/release/update-version.sh
@@ -62,4 +62,7 @@ for FILE in .github/workflows/*.yaml; do
   sed_runner "/shared-workflows/ s/@.*/@branch-${NEXT_SHORT_TAG}/g" "${FILE}"
 done
 
-sed_runner "s/branch-[0-9][0-9].[0-9][0-9]/branch-${NEXT_SHORT_TAG}/" ./docs/cugraph-docs/source/nx_cugraph/nx_cugraph.md
+# Update container_image version in workflow files
+for FILE in .github/workflows/*.yaml; do
+  sed_runner "s/rapidsai\/ci-conda:[0-9][0-9].[0-9][0-9]-latest/rapidsai\/ci-conda:${NEXT_SHORT_TAG}-latest/g" "${FILE}"
+done