diff --git a/ci/docker/ubuntu-gnu-cpp.dockerfile b/ci/docker/ubuntu-gnu-cpp.dockerfile
index b80ee7abac..44f2499ac1 100644
--- a/ci/docker/ubuntu-gnu-cpp.dockerfile
+++ b/ci/docker/ubuntu-gnu-cpp.dockerfile
@@ -68,12 +68,8 @@ RUN ./cmake.sh 3.23.4 ${arch}
 ENV PATH=/cmake/bin/:$PATH
 ENV LESSCHARSET=utf-8
 
-ARG VT_TV_ENABLED
-
-RUN echo "VT_TV_ENABLED is set to: $VT_TV_ENABLED"
-
 COPY ./ci/deps/vtk.sh vtk.sh
-RUN if [ "$VT_TV_ENABLED" ]; then \
+RUN if test ${vt_tv_enabled} -eq 1; then \
       chmod +x vtk.sh && \
       ./vtk.sh 9.3.0 /vtk-build -j4; \
     fi
diff --git a/docker-compose.yml b/docker-compose.yml
index ded43a85bc..645364da8a 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -137,6 +137,7 @@ services:
         host_compiler: ${HOST_COMPILER}
         ubuntu: ${UBUNTU}
         ubsan_enabled: ${VT_UBSAN:-0}
+        vt_tv_enabled: ${VT_TV_ENABLED:-0}
         zoltan_enabled: ${VT_ZOLTAN:-0}
         external_fmt: ${VT_EXTERNAL_FMT:-0}
       cache_from: