diff --git a/test/go/genmock.sh b/test/go/genmock.sh
index 9bd2a3c..687ce61 100644
--- a/test/go/genmock.sh
+++ b/test/go/genmock.sh
@@ -4,8 +4,8 @@
 
 export GOPATH=$(mktemp -d -t gopath-XXXXXXXXXX)
 
-go install -mod=mod github.com/golang/mock/mockgen
+go install github.com/golang/mock/mockgen
 
-`go env GOPATH`/bin/mockgen -build_flags "-mod=mod" -destination=src/common/mock_handler.go -package=common github.com/apache/thrift/test/go/src/gen/thrifttest ThriftTest
+`go env GOPATH`/bin/mockgen -destination=src/common/mock_handler.go -package=common github.com/apache/thrift/test/go/src/gen/thrifttest ThriftTest
 
 chmod a+w -R $GOPATH && rm -Rf $GOPATH