diff --git a/.travis.yml b/.travis.yml
index 90a007a8..e9abbed8 100644
--- a/.travis.yml
+++ b/.travis.yml
@@ -20,7 +20,7 @@ jobs:
 
     - stage: build
       name: build docker image
-      script: bash scripts/image_build.sh $TRAVIS_REPO_SLUG
+      script: bash scripts/image_build.sh $TRAVIS_REPO_SLUG $TRAVIS_BRANCH
       if: env(CD_BUILD_IGNORE) IS NOT present AND type NOT IN (pull_request) AND env(CD_DOCKER_USERNAME) IS present AND env(CD_DOCKER_PASSWORD) IS present
 
 notifications:
diff --git a/Dockerfile b/Dockerfile
index bd23775d..33a873c6 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -6,13 +6,15 @@ RUN apt-get update -qq && apt-get install -y build-essential libpq-dev nodejs
 # Set an environment variable for the install location.
 ENV RAILS_ROOT /usr/src/app
 
+# Sets the footer of greenlight application with current build version
+ARG version_code
+ENV VERSION_CODE=$version_code
+
 # Make the directory and set as working.
 RUN mkdir -p $RAILS_ROOT
 WORKDIR $RAILS_ROOT
 
-# Set environment variables.
-ARG version_code
-ENV VERSION_CODE=$version_code
+# Set Rails environment.
 ENV RAILS_ENV production
 
 # Adding project files.