diff --git a/docker/Dockerfile-base b/docker/Dockerfile-base
index b2308b02a..ab118cef9 100644
--- a/docker/Dockerfile-base
+++ b/docker/Dockerfile-base
@@ -4,8 +4,9 @@ ARG UBUNTU_VERSION="22.04"
 ARG MAX_JOBS=4
 
 ARG BASE_IMAGE=nvidia/cuda
+ARG DEFAULT_TAG=${CUDA_VERSION}-cudnn${CUDNN_VERSION}-devel-ubuntu${UBUNTU_VERSION}
 ARG BASE_TAG=""
-FROM $BASE_IMAGE:${BASE_TAG:-$CUDA_VERSION-cudnn$CUDNN_VERSION-devel-ubuntu$UBUNTU_VERSION} AS base-builder
+FROM ${BASE_IMAGE}:${BASE_TAG:-${DEFAULT_TAG}} AS base-builder
 
 ENV PATH="/root/miniconda3/bin:${PATH}"