diff --git a/.github/workflows/docs-ci.yaml b/.github/workflows/docs-ci.yaml
index 9915f33..3592bda 100644
--- a/.github/workflows/docs-ci.yaml
+++ b/.github/workflows/docs-ci.yaml
@@ -37,7 +37,9 @@ jobs:
       GH_TOKEN: ${{ secrets.GH_TOKEN_NVIDIA_CI_CD }}
       TAG: ${{ inputs.git_tag }}
       REF_NAME: ${{ github.ref_name }}
-      REF_TYPE: ${{ github.ref_type }}       
+      REF_TYPE: ${{ github.ref_type }}
+      REF_HEAD: ${{ github.head_ref }}
+      PR_NUMBER: ${{ github.event.number }}  
       PR_TITLE_PREFIX: "task: update documentation for"
     steps:
     - uses: actions/checkout@v4
@@ -52,7 +54,14 @@ jobs:
       run: |
         echo "REF_TYPE:" $REF_TYPE
         echo "REF_NAME:" $REF_NAME
-        BRANCH=$REF_NAME
+        echo "REF_HEAD:" $REF_HEAD
+        if [[ -n $PR_NUMBER ]]; then
+          echo PR_NUMBER=$PR_NUMBER
+        fi
+
+        #BRANCH=$REF_NAME
+        git pull
+        git checkout test-netops-docs-workflow #$REF_HEAD        
         make api-docs helm-docs generate-docs-versions-var
     - name: Close any existing documentation PRs
       run: |
@@ -74,7 +83,7 @@ jobs:
         gh repo sync $DOWNSTREAM_REPO_OWNER/network-operator-docs --source $UPSTREAM_REPO_OWNER/network-operator-docs --branch $UPSTREAM_DEFAULT_BRANCH
 
         git pull
-        git checkout test-netops-docs-workflow
+        git checkout test-netops-docs-workflow #$REF_HEAD
         git checkout -b $DOWNSTREAM_FEATURE_BRANCH
         git status
         git add docs
diff --git a/Makefile b/Makefile
index ce64227..20a363a 100644
--- a/Makefile
+++ b/Makefile
@@ -19,8 +19,8 @@ SRC := $(shell echo $(if $(TAG),$(TAG),$(BRANCH)) | sed 's/^v//')
 
 # Network Operator source tar location
 #REPO_TAR_URL ?= https://github.com/Mellanox/network-operator/archive/refs/$(if $(TAG),tags/$(TAG),heads/$(BRANCH)).tar.gz
-REPO_TAR_URL = https://github.com/heyvister1/network-operator/archive/refs/$(if $(TAG),tags/$(TAG),heads/$(BRANCH)).tar.gz
-
+REPO_TAR_URL = https://github.com/heyvister1/network-operator/archive/refs/$(if $(TAG),tags/$(TAG),pull/$(PR_NUMBER))/head.tar.gz
+#REPO_TAR_URL = github.com/heyvister1/network-operator/archive/refs/pull/2/head.tar.gz
 # release.yaml location
 #RELEASE_YAML_URL ?= https://raw.githubusercontent.com/Mellanox/network-operator/$(if $(TAG),$(TAG),$(BRANCH))/hack/release.yaml
 RELEASE_YAML_URL = https://raw.githubusercontent.com/heyvister1/network-operator/$(if $(TAG),$(TAG),$(BRANCH))/hack/release.yaml