diff --git a/.github/workflows/doc-build.yml b/.github/workflows/doc-build.yml
index 6ec2607ae2f..268bfbd4cd3 100644
--- a/.github/workflows/doc-build.yml
+++ b/.github/workflows/doc-build.yml
@@ -109,8 +109,10 @@ jobs:
 
         # Convert refs/tags/v1.12.0rc3 into 1.12.
         # Adopted from https://github.com/pytorch/pytorch/blob/main/.github/workflows/_docs.yml#L150C11-L155C13
-        if [[ "${GITHUB_REF}" =~ ^refs/tags/v([0-9]+\.[0-9]+) ]] || [[ "${GITHUB_REF}" == "refs/heads/release/0.6" ]]; then
+        if [[ "${GITHUB_REF}" =~ ^refs/tags/v([0-9]+\.[0-9]+) ]]; then
           TARGET_FOLDER="${BASH_REMATCH[1]}"
+        elif [[ "${GITHUB_REF}" == "refs/heads/release/0.6" ]]; then
+          TARGET_FOLDER="0.6"
         else
           TARGET_FOLDER="main"
         fi