diff --git a/.devops/llama-server-cuda.Dockerfile b/.devops/llama-server-cuda.Dockerfile
index e21d7941b..fb52d573d 100644
--- a/.devops/llama-server-cuda.Dockerfile
+++ b/.devops/llama-server-cuda.Dockerfile
@@ -1,6 +1,5 @@
 ARG UBUNTU_VERSION=20.04
-ARG DEBIAN_FRONTEND=noninteractive
-ENV TZ=Africa/Johannesburg 
+
 # This needs to generally match the container host's environment.
 ARG CUDA_VERSION=11.3.1
 # Target the CUDA build image
@@ -9,7 +8,8 @@ ARG BASE_CUDA_DEV_CONTAINER=nvidia/cuda:${CUDA_VERSION}-devel-ubuntu${UBUNTU_VER
 ARG BASE_CUDA_RUN_CONTAINER=nvidia/cuda:${CUDA_VERSION}-runtime-ubuntu${UBUNTU_VERSION}
 
 FROM ${BASE_CUDA_DEV_CONTAINER} AS build
-
+ARG DEBIAN_FRONTEND=noninteractive
+ENV TZ=Africa/Johannesburg 
 # CUDA architecture to build for (defaults to all supported archs)
 ARG CUDA_DOCKER_ARCH=default