diff --git a/docker/Dockerfile.vk b/docker/Dockerfile.vk
index 9a455a24..f0249d25 100644
--- a/docker/Dockerfile.vk
+++ b/docker/Dockerfile.vk
@@ -7,7 +7,6 @@ WORKDIR /app
 COPY .. .
 
 ARG VERSION
-ENV KUBELET_VERSION=${VERSION}
 
 ENV GOMODCACHE="/go/pkg/mod"
 ENV GOCACHE="/go/build-cache"
@@ -15,7 +14,7 @@ ENV GOCACHE="/go/build-cache"
 RUN mkdir -p $GOMODCACHE && mkdir -p $GOCACHE
 
 
-RUN bash -c "./cmd/virtual-kubelet/set-version.sh"
+RUN bash -c "KUBELET_VERSION=${VERSION} ./cmd/virtual-kubelet/set-version.sh"
 RUN go mod tidy
 RUN CGO_ENABLED=0 GOOS=linux go build -o bin/vk cmd/virtual-kubelet/main.go