diff --git a/.github/workflows/test-build-push-image.yml b/.github/workflows/test-build-push-image.yml
index 5922e18c..3ec0ce46 100644
--- a/.github/workflows/test-build-push-image.yml
+++ b/.github/workflows/test-build-push-image.yml
@@ -13,7 +13,7 @@ jobs:
       id-token: 'write'
     with:
       region: us-central1
-      dockerfile_path: ./containers/pytorch/2.1/transformers/4.37.2/py310/Dockerfile
+      dockerfile_path: ./containers/pytorch/training/gpu/2.1/transformers/4.37.2/py310/Dockerfile
       docker_image_tag: huggingface-pytorch-transformers-4.37.2-py310:latest
       gcp_artifact_registry_repository: deep-learning-images
     secrets: