diff --git a/ecs/Dockerfile b/ecs/Dockerfile
index 8d18ddfed..dead5829d 100644
--- a/ecs/Dockerfile
+++ b/ecs/Dockerfile
@@ -18,6 +18,8 @@ COPY . .
 FROM base AS make-plugin
 ARG TARGETOS
 ARG TARGETARCH
+RUN apk add build-base
+RUN GO111MODULE=on go get github.com/golang/mock/mockgen@latest
 RUN --mount=type=cache,target=/root/.cache/go-build \
     --mount=type=cache,target=/go/pkg/mod \
     GOOS=${TARGETOS} \
diff --git a/ecs/builder.Makefile b/ecs/builder.Makefile
index 8b6920baa..18bd2da24 100644
--- a/ecs/builder.Makefile
+++ b/ecs/builder.Makefile
@@ -20,7 +20,10 @@ all: build
 clean:
 	rm -rf dist/
 
-build:
+generate: pkg/amazon/sdk/api_mock.go
+	go generate ./...
+
+build: generate
 	$(GO_BUILD) -v -o $(BINARY_WITH_EXTENSION) cmd/main/main.go
 
 cross: