diff --git a/Makefile b/Makefile
index af657a66d..70d642412 100644
--- a/Makefile
+++ b/Makefile
@@ -87,7 +87,7 @@ export XC_ARCH
 ARCH:=${XC_OS}_${XC_ARCH}
 export ARCH
 
-export DBUILD_ARGS=--build-arg DBUILD_DATE=${DBUILD_DATE} --build-arg DBUILD_REPO_URL=${DBUILD_REPO_URL} --build-arg DBUILD_SITE_URL=${DBUILD_SITE_URL}
+export DBUILD_ARGS=--build-arg DBUILD_DATE=${DBUILD_DATE} --build-arg DBUILD_REPO_URL=${DBUILD_REPO_URL} --build-arg DBUILD_SITE_URL=${DBUILD_SITE_URL} --build-arg BRANCH=${BRANCH} --build-arg RELEASE_TAG=${RELEASE_TAG}
 
 
 # Specify the name for the binary
diff --git a/buildscripts/cstor-csi-driver/cstor-csi-driver.Dockerfile b/buildscripts/cstor-csi-driver/cstor-csi-driver.Dockerfile
index 82320bf49..437d16aac 100644
--- a/buildscripts/cstor-csi-driver/cstor-csi-driver.Dockerfile
+++ b/buildscripts/cstor-csi-driver/cstor-csi-driver.Dockerfile
@@ -14,6 +14,8 @@
 
 FROM golang:1.14.7 as build
 
+ARG BRANCH
+ARG RELEASE_TAG
 ARG TARGETOS
 ARG TARGETARCH
 ARG TARGETVARIANT=""
@@ -23,7 +25,9 @@ ENV GO111MODULE=on \
   GOARCH=${TARGETARCH} \
   GOARM=${TARGETVARIANT} \
   DEBIAN_FRONTEND=noninteractive \
-  PATH="/root/go/bin:${PATH}"
+  PATH="/root/go/bin:${PATH}" \
+  BRANCH=${BRANCH} \
+  RELEASE_TAG=${RELEASE_TAG}
 
 WORKDIR /go/src/github.com/openebs/cstor-csi/