From 7a74b59a4fa361bfd687e7e7326bb1db0d684231 Mon Sep 17 00:00:00 2001
From: Drew Larson <drew@drwlrsn.com>
Date: Wed, 28 Jun 2017 21:14:48 -0600
Subject: [PATCH] Better branch checking.

---
 .travis.yml | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/.travis.yml b/.travis.yml
index 9763939..b222b45 100644
--- a/.travis.yml
+++ b/.travis.yml
@@ -7,7 +7,9 @@ branches:
   only:
   - master
 after_success:
-  - if [ "$TRAVIS_BRANCH" == "master" ]; then
+  - BRANCH=${TRAVIS_PULL_REQUEST_BRANCH:-$TRAVIS_BRANCH}
+  - echo $BRANCH
+  - if [ "$BRANCH" == "master" ]; then
     docker login -u="$DOCKER_USERNAME" -p="$DOCKER_PASSWORD";
     docker push bcbc/workstand:production;
     fi