diff --git a/.github/workflows/cicd.yml b/.github/workflows/cicd.yml
index dbc247c94c..5e1e5aa6bb 100644
--- a/.github/workflows/cicd.yml
+++ b/.github/workflows/cicd.yml
@@ -7,9 +7,7 @@ jobs:
     runs-on: ubuntu-latest
     container: ultimaker/cura-build-environment
     steps:
-    - name: Checkout master
-      uses: actions/checkout@v1
+    - name: Checkout Cura
+      uses: actions/checkout@v2
     - name: Build and test
-      env:
-        GIT_REF_NAME: github.ref
       run: docker/build.sh
diff --git a/docker/build.sh b/docker/build.sh
index a772c00509..4510692bf2 100755
--- a/docker/build.sh
+++ b/docker/build.sh
@@ -20,7 +20,14 @@ cd "${PROJECT_DIR}"
 # Check the branch to use:
 #  1. Use the Uranium branch with the branch same if it exists.
 #  2. Otherwise, use the default branch name "master"
-echo "GIT_REF_NAME: ${GIT_REF_NAME}"
+echo "GITHUB_REF: ${GITHUB_REF}"
+echo "GITHUB_BASE_REF: ${GITHUB_BASE_REF}"
+
+GIT_REF_NAME="${GITHUB_REF}"
+if [ -n "${GITHUB_BASE_REF}" ]; then
+  GIT_REF_NAME="${GITHUB_BASE_REF}"
+fi
+
 URANIUM_BRANCH="${GIT_REF_NAME:-master}"
 output="$(git ls-remote --heads https://github.com/Ultimaker/Uranium.git "${URANIUM_BRANCH}")"
 if [ -z "${output}" ]; then