diff --git a/bench/proxy/Dockerfile b/bench/proxy/Dockerfile
index 4608bf1b..2f6cd680 100644
--- a/bench/proxy/Dockerfile
+++ b/bench/proxy/Dockerfile
@@ -12,6 +12,7 @@ COPY go.mod go.sum ./
 RUN go mod download
 
 COPY . ./
+RUN go generate ./...
 RUN go build -o /mediamtx .
 
 COPY bench/proxy/start.sh /
diff --git a/bench/publish/Dockerfile b/bench/publish/Dockerfile
index 442b913f..035d0cd3 100644
--- a/bench/publish/Dockerfile
+++ b/bench/publish/Dockerfile
@@ -12,6 +12,7 @@ COPY go.mod go.sum ./
 RUN go mod download
 
 COPY . ./
+RUN go generate ./...
 RUN go build -o /mediamtx .
 
 COPY bench/publish/start.sh /
diff --git a/bench/read/Dockerfile b/bench/read/Dockerfile
index 35a6fb95..baecc75d 100644
--- a/bench/read/Dockerfile
+++ b/bench/read/Dockerfile
@@ -12,6 +12,7 @@ COPY go.mod go.sum ./
 RUN go mod download
 
 COPY . ./
+RUN go generate ./...
 RUN go build -o /mediamtx .
 
 COPY bench/read/start.sh /