diff --git a/docker/Dockerfile.ci.dev b/docker/Dockerfile.ci.dev
index 377bd4a2005..bd898d10166 100644
--- a/docker/Dockerfile.ci.dev
+++ b/docker/Dockerfile.ci.dev
@@ -1,8 +1,8 @@
 # Copyright (c) 2025, NVIDIA CORPORATION. All rights reserved.
 # syntax=docker/dockerfile:1.3-labs
 
-ARG FROM_IMAGE_NAME
-FROM ${FROM_IMAGE_NAME} as main
+ARG FROM_IMAGE_NAME=nvcr.io/nvidia/pytorch:26.04-py3
+FROM ${FROM_IMAGE_NAME} AS main
 ENV PIP_CONSTRAINT=""
 ENV DEBIAN_FRONTEND=noninteractive
 ARG UV_VERSION=0.7.2
@@ -82,7 +82,7 @@ COPY assets/ /opt/data/
 ENV UV_PYTHON=$UV_PROJECT_ENVIRONMENT/bin/python
 
 ##### For NVIDIANS only #####
-FROM main as jet
+FROM main AS jet
 ARG JET_API_VERSION
 ENV PATH="$PATH:/opt/jet/bin"
 RUN --mount=type=secret,id=JET_INDEX_URLS bash -ex <<"EOF"
diff --git a/docker/Dockerfile.ci.nemo b/docker/Dockerfile.ci.nemo
index b00349e101a..8f250b4374a 100644
--- a/docker/Dockerfile.ci.nemo
+++ b/docker/Dockerfile.ci.nemo
@@ -1,8 +1,8 @@
 # Copyright (c) 2025, NVIDIA CORPORATION. All rights reserved.
 # syntax=docker/dockerfile:1.3-labs
 
-ARG FROM_IMAGE_NAME
-FROM ${FROM_IMAGE_NAME} as main
+ARG FROM_IMAGE_NAME=nvcr.io/nvidia/pytorch:26.04-py3
+FROM ${FROM_IMAGE_NAME} AS main
 
 RUN apt-get update && \
     apt-get install -y --no-install-recommends gettext && \
@@ -11,7 +11,7 @@ RUN apt-get update && \
     chmod a+x /usr/local/bin/yq
 
 ##### For NVIDIANS only #####
-FROM main as jet
+FROM main AS jet
 ARG JET_API_VERSION
 RUN --mount=type=secret,id=JET_INDEX_URLS \
     JET_INDEX_URLS=$(cat /run/secrets/JET_INDEX_URLS) && \
diff --git a/docker/Dockerfile.linting b/docker/Dockerfile.linting
index 259c0bbedcd..bf27b768374 100644
--- a/docker/Dockerfile.linting
+++ b/docker/Dockerfile.linting
@@ -1,7 +1,7 @@
 # syntax=docker/dockerfile:experimental
 
-ARG FROM_IMAGE_NAME
-FROM $FROM_IMAGE_NAME as main
+ARG FROM_IMAGE_NAME=nvcr.io/nvidia/pytorch:26.04-py3
+FROM $FROM_IMAGE_NAME AS main
 ENV DEBIAN_FRONTEND=noninteractive
 ARG UV_VERSION=0.7.2
 ARG YQ_VERSION=4.44.1
@@ -16,7 +16,7 @@ COPY megatron/core/package_info.py megatron/core/__init__.py /opt/megatron-lm/me
 RUN uv sync --locked --only-group linting --only-group test --only-group ci
 
 ##### For NVIDIANS only #####
-FROM main as jet
+FROM main AS jet
 ARG JET_API_VERSION
 RUN --mount=type=secret,id=JET_INDEX_URLS \
       JET_INDEX_URLS=$(cat /run/secrets/JET_INDEX_URLS) && \ 
diff --git a/examples/multimodal/Dockerfile b/examples/multimodal/Dockerfile
index d7c4fd41af5..46db1319566 100644
--- a/examples/multimodal/Dockerfile
+++ b/examples/multimodal/Dockerfile
@@ -37,7 +37,7 @@ RUN uv pip install --system --no-cache --break-system-packages \
     flask-restful \
     wandb \
     bitstring \
-    filetype
+    filetype \
     setuptools
 
 # Install CLIP from GitHub