diff --git a/battle_tests/Dockerfile b/battle_tests/Dockerfile
index 1e449c806..a0b9afffc 100644
--- a/battle_tests/Dockerfile
+++ b/battle_tests/Dockerfile
@@ -3,6 +3,8 @@ FROM golang:latest AS builder
 
 ARG CURIO_VERSION
 ARG BATTLE_TEST_SHA
+ARG ATTEMPT
+ARG LANGUAGE
 
 RUN mkdir /app
 ADD . /app/
@@ -10,7 +12,7 @@ WORKDIR /app
 
 # RUN apk update && apk upgrade && apk add --no-cache build-base git
 
-RUN go build -ldflags="-X 'github.com/bearer/curio/battle_tests/build.CurioVersion=${CURIO_VERSION}' -X 'github.com/bearer/curio/battle_tests/build.BattleTestSHA=${BATTLE_TEST_SHA}' -X 'github.com/bearer/curio/battle_tests/build.Attempt=${ATTEMPT}' -X 'github.com/bearer/curio/battle_tests/build.Language=${Language}'" \
+RUN go build -ldflags="-X 'github.com/bearer/curio/battle_tests/build.CurioVersion=${CURIO_VERSION}' -X 'github.com/bearer/curio/battle_tests/build.BattleTestSHA=${BATTLE_TEST_SHA}' -X 'github.com/bearer/curio/battle_tests/build.Attempt=${ATTEMPT}' -X 'github.com/bearer/curio/battle_tests/build.Language=${LANGUAGE}'" \
   -a -o dist/battle_tests ./battle_tests/battle_tests.go
 
 # FROM alpine