diff --git a/.travis.yml b/.travis.yml
index 2142aef..630a370 100644
--- a/.travis.yml
+++ b/.travis.yml
@@ -18,11 +18,9 @@ deploy:
 after_deploy:
   - docker login -e $DOCKER_EMAIL -u $DOCKER_USER -p $DOCKER_PASS
   - export REPO=gettyimages/marathon_exporter
-  - export TAG=`if [ "$TRAVIS_BRANCH" == "master" ]; then echo "latest"; else echo $TRAVIS_BRANCH ; fi`
-  - docker build -f Dockerfile -t $REPO:$COMMIT .
-  - docker tag $REPO:$COMMIT $REPO:$TAG
-  - docker push $REPO:$TAG
-  - docker tag $REPO:$COMMIT $REPO:latest
+  - docker build -f Dockerfile -t $REPO:$TRAVIS_TAG .
+  - docker push $REPO:$TRAVIS_TAG
+  - docker tag $REPO:$TRAVIS_TAG $REPO:latest
   - docker push $REPO:latest
 
 env: