diff --git a/.github/workflows/Deploy.yml b/.github/workflows/Deploy.yml
index 52cce89..21888e8 100644
--- a/.github/workflows/Deploy.yml
+++ b/.github/workflows/Deploy.yml
@@ -90,13 +90,13 @@ jobs:
       - name: Build and push Docker image
         uses: docker/build-push-action@v3
         with:
-          context: ./modules/ml-pipeline/src/pipeline
+          context: .
           file: ./deployment/Dockerfile.prediction.lambda
           push: true
           tags: ${{ steps.set_ecr_credentials.outputs.ecr_uri }}:${{ github.sha }}
           cache-from: type=gha
           cache-to: type=gha,mode=max
-          platform: linux/amd64
+          platforms: linux/amd64
           provenance: false
           build-args: |
             RUNTIME_ENVIRONMENT=${{ steps.set_runtime_environment.outputs.runtime_environment }}