diff --git a/examples/gke-a4x-max-bm/nccl-jobset-example.yaml b/examples/gke-a4x-max-bm/nccl-jobset-example.yaml
index 77fd396720..cb8d15bb68 100644
--- a/examples/gke-a4x-max-bm/nccl-jobset-example.yaml
+++ b/examples/gke-a4x-max-bm/nccl-jobset-example.yaml
@@ -78,7 +78,7 @@ spec:
             - name: nccl-test
               stdin: true
               tty: true
-              image: us-docker.pkg.dev/gce-ai-infra/gpudirect-gib/nccl-gib-a4x-max-arm64:v1.1.1
+              image: us-docker.pkg.dev/gce-ai-infra/gpudirect-gib/nccl-plugin-gib-diagnostic-arm64:v1.1.2
               securityContext:
                 capabilities:
                   add: ["IPC_LOCK"]