diff --git a/hooks/build b/hooks/build
index e8385d6..0b09918 100755
--- a/hooks/build
+++ b/hooks/build
@@ -5,11 +5,10 @@
 echo "Build hook running"
 echo "IMAGE_NAME = ${IMAGE_NAME}"
 echo "DOCKER_REPO = ${DOCKER_REPO}"
-echo "CACHE_TAG = ${CACHE_TAG}"
-export BRANCH=${TRAVIS_BRANCH:-`git symbolic-ref --short HEAD`}
+export BRANCH=${SOURCE_BRANCH:-`git symbolic-ref --short HEAD`}
 
 export DATE=`date -u +"%Y-%m-%dT%H:%M:%SZ"`
-export COMMIT=${TRAVIS_COMMIT:-`git rev-parse --short HEAD`}
+export COMMIT=${SOURCE_COMMIT:-`git rev-parse --short HEAD`}
 docker build --build-arg BUILD_DATE=$DATE \
              --build-arg VCS_REF=$COMMIT \
              --build-arg BRANCH=$BRANCH \