diff --git a/gen.go b/gen.go
index 9b73354abf2..65f0a7782dd 100644
--- a/gen.go
+++ b/gen.go
@@ -2,4 +2,4 @@ package sourcegraph
 
 // Keep these versions in sync with go.mod
 //go:generate env GOBIN=$PWD/.bin GO111MODULE=on go install golang.org/x/tools/cmd/goimports@v0.17.0
-//go:generate go run github.com/strum355/go-mockgen/v2/cmd/go-mockgen@v2.0.0-20240306201845-b2e8d553a343
+//go:generate go run github.com/derision-test/go-mockgen/v2/cmd/go-mockgen@v2.0.1