Fix docker build by using the build-arg BRANCH