diff --git a/demos/golang/grpc_benchmark/download_and_build_grpc_benchmark.sh b/demos/golang/grpc_benchmark/download_and_build_grpc_benchmark.sh
index 2a510d0d..3da654aa 100755
--- a/demos/golang/grpc_benchmark/download_and_build_grpc_benchmark.sh
+++ b/demos/golang/grpc_benchmark/download_and_build_grpc_benchmark.sh
@@ -1,10 +1,14 @@
 #!/bin/bash
 set -e
 
-export GOPATH=$PWD
+export GOPATH=$HOME/go
 out_dir=$PWD/bin
-occlum-go get -u google.golang.org/grpc
-cd src/google.golang.org/grpc/
 rm -rf ${out_dir}
 mkdir ${out_dir}
-occlum-go build -o ${out_dir}/server $GOPATH/src/google.golang.org/grpc/benchmark/server/main.go && occlum-go build -o ${out_dir}/client $GOPATH/src/google.golang.org/grpc/benchmark/client/main.go
+rm -f go.mod
+occlum-go mod init grpc_benchmark
+occlum-go mod tidy
+occlum-go get -u google.golang.org/grpc@v1.50.1
+cd ${GOPATH}/pkg/mod/google.golang.org/grpc@v1.50.1
+occlum-go build -o ${out_dir}/server ./benchmark/server/main.go
+occlum-go build -o ${out_dir}/client ./benchmark/client/main.go