diff --git a/ops/buildkite/build-code b/ops/buildkite/build-code
index bc34a9efe27..66ccbaaada1 100755
--- a/ops/buildkite/build-code
+++ b/ops/buildkite/build-code
@@ -25,7 +25,7 @@ build_code () {
     echo "--- Build code from branch master"
 
     # TODO: uncomment this to avoid deploys from non-master branch
-    # if [[ "$BUILDKITE_BRANCH" != "master" ]]; then
+    # if [[ "$BRANCH" != "master" ]]; then
     #     echo "Error: You're not on the master branch. Exiting."
     #     exit 1
     # fi
@@ -55,8 +55,8 @@ update_owid_grapher_repo() {
     (
         cd owid-grapher
         git fetch --all -q
-        git checkout "$BUILDKITE_BRANCH" -q
-        git reset --hard origin/"$BUILDKITE_BRANCH"
+        git checkout "$BRANCH" -q
+        git reset --hard origin/"$BRANCH"
     )
 }