diff --git a/.github/workflows/test-workflow.yaml b/.github/workflows/test-workflow.yaml
index 7702dc9..911b883 100644
--- a/.github/workflows/test-workflow.yaml
+++ b/.github/workflows/test-workflow.yaml
@@ -18,12 +18,11 @@ jobs:
         run: |
           version=""
           if [[ "${{ github.event_name }}" == "push" ]]; then
-            version="${{ github.event.push.base.ref }}"
+            version="$GITHUB_REF_NAME"
             echo "Version: $version"
             echo "Ref_name: ${{ github.event.push.base.ref_name }}"
           else
-            version=${{ github.event.inputs.branch }}
+            version=${{ inputs.branch }}
             echo "Version: $version"
-            echo "Ref: ${{ github.event.push.base.ref }}"
           fi
           env | grep sundar