diff --git a/.github/workflows/docker-unified.yml b/.github/workflows/docker-unified.yml
index 2ccda4e088e70..1570dcc6daac9 100644
--- a/.github/workflows/docker-unified.yml
+++ b/.github/workflows/docker-unified.yml
@@ -892,7 +892,7 @@ jobs:
       - name: Check out the repo
         uses: actions/checkout@v2
         with:
-          fetch-depth: 0
+          fetch-depth: 800
       - name: Build and push
         uses: ./.github/actions/docker-custom-build-and-push
         with:
diff --git a/docker/ingest-api/Dockerfile b/docker/ingest-api/Dockerfile
index 45e8cede085c7..b4e636185e5ff 100644
--- a/docker/ingest-api/Dockerfile
+++ b/docker/ingest-api/Dockerfile
@@ -5,7 +5,7 @@ RUN apt update && apt install -y build-essential python3-dev
 # FROM tiangolo/uvicorn-gunicorn-fastapi:python3.8-slim as base
 FROM openjdk:8 as prod-build
 COPY . /datahub-src
-RUN cd /datahub-src && ./gradlew :metadata-events:mxe-schemas:build
+RUN cd /datahub-src && ./gradlew :metadata-ingestion:codegen
 
 FROM base as prod-codegen
 COPY --from=prod-build /datahub-src /datahub-src