diff --git a/test/test_artifacts/v0/gpu-dependencies.test.Dockerfile b/test/test_artifacts/v0/gpu-dependencies.test.Dockerfile
index 2e48e09d..f008e6fc 100644
--- a/test/test_artifacts/v0/gpu-dependencies.test.Dockerfile
+++ b/test/test_artifacts/v0/gpu-dependencies.test.Dockerfile
@@ -8,4 +8,4 @@ ARG MAMBA_DOCKERFILE_ACTIVATE=1
 # 2. Check if Pytorch is installed with CUDA support for GPU image
 COPY --chown=$MAMBA_USER:$MAMBA_USER scripts/cuda_validation.py .
 RUN chmod +x cuda_validation.py
-RUN python3 cuda_validation.py
\ No newline at end of file
+CMD python3 cuda_validation.py
\ No newline at end of file
diff --git a/test/test_artifacts/v1/gpu-dependencies.test.Dockerfile b/test/test_artifacts/v1/gpu-dependencies.test.Dockerfile
index 2e48e09d..f008e6fc 100644
--- a/test/test_artifacts/v1/gpu-dependencies.test.Dockerfile
+++ b/test/test_artifacts/v1/gpu-dependencies.test.Dockerfile
@@ -8,4 +8,4 @@ ARG MAMBA_DOCKERFILE_ACTIVATE=1
 # 2. Check if Pytorch is installed with CUDA support for GPU image
 COPY --chown=$MAMBA_USER:$MAMBA_USER scripts/cuda_validation.py .
 RUN chmod +x cuda_validation.py
-RUN python3 cuda_validation.py
\ No newline at end of file
+CMD python3 cuda_validation.py
\ No newline at end of file