diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
index 8d223b1..97b3466 100644
--- a/.github/workflows/test.yml
+++ b/.github/workflows/test.yml
@@ -54,7 +54,7 @@ jobs:
           python-version: ${{ env.PYTHON_VERSION }}
       - name: Setup environment
         run: |     
-          go get github.com/testit-tms/adapters-go/pkg/tms@${{ GITHUB_HEAD_REF }}
+          go get github.com/testit-tms/adapters-go/pkg/tms@${GITHUB_HEAD_REF:-${GITHUB_REF#refs/heads/}}
           dotnet build --configuration Release --property WarningLevel=0 api-validator-dotnet
           pip install testit-cli
       - name: Create TestRun