diff --git a/.github/workflows/publish-example-images.yaml b/.github/workflows/publish-example-images.yaml
index dfe2ae9004..7bbf3e2170 100644
--- a/.github/workflows/publish-example-images.yaml
+++ b/.github/workflows/publish-example-images.yaml
@@ -38,8 +38,9 @@ jobs:
             dockerfile: examples/pytorch/elastic/imagenet/Dockerfile
           - component-name: pytorch-elastic-example-echo
             dockerfile: examples/pytorch/elastic/echo/Dockerfile
-          - component-name: pytorch-cpu-demo-torchrun-cpu
+          - component-name: pytorch-torchrun-cpu
             dockerfile: examples/pytorch/cpu-demo/Dockerfile
+            context: examples/pytorch/cpu-demo
           - component-name: xgboost-dist-iris
             dockerfile: examples/xgboost/xgboost-dist/Dockerfile
             context: examples/xgboost/xgboost-dist