diff --git a/doc/COMPILE.md b/doc/COMPILE.md
index abb66084ac6f6c57c13c940eb10a87e2aba2daa2..640d599dda555a8fef70ee0c42de29eae022c720 100644
--- a/doc/COMPILE.md
+++ b/doc/COMPILE.md
@@ -75,6 +75,8 @@ export PATH=$PATH:$GOPATH/bin
 ## Get go packages
 
 ```shell
+go env -w GO111MODULE=on
+go env -w GOPROXY=https://goproxy.cn,direct
 go get -u github.com/grpc-ecosystem/grpc-gateway/protoc-gen-grpc-gateway
 go get -u github.com/grpc-ecosystem/grpc-gateway/protoc-gen-swagger
 go get -u github.com/golang/protobuf/protoc-gen-go
diff --git a/doc/COMPILE_CN.md b/doc/COMPILE_CN.md
index 2ddaaf71f23b0199c7458d068139a6b7169c25d8..392da7ed64bc88a8b92294f2a1f805522433cad1 100644
--- a/doc/COMPILE_CN.md
+++ b/doc/COMPILE_CN.md
@@ -72,6 +72,8 @@ export PATH=$PATH:$GOPATH/bin
 ## 获取 Go packages
 
 ```shell
+go env -w GO111MODULE=on
+go env -w GOPROXY=https://goproxy.cn,direct
 go get -u github.com/grpc-ecosystem/grpc-gateway/protoc-gen-grpc-gateway
 go get -u github.com/grpc-ecosystem/grpc-gateway/protoc-gen-swagger
 go get -u github.com/golang/protobuf/protoc-gen-go
diff --git a/python/pipeline/gateway/proxy_server.go b/python/pipeline/gateway/proxy_server.go
index a74e798463b58efe26ab027c649a07131d4bbf32..cadc1567bca60de13970d7dd03481ec103226f47 100644
--- a/python/pipeline/gateway/proxy_server.go
+++ b/python/pipeline/gateway/proxy_server.go
@@ -25,7 +25,7 @@ import (
   "github.com/grpc-ecosystem/grpc-gateway/runtime"
   "google.golang.org/grpc"
 
-  gw "./proto"
+  gw "serving-gateway/proto"
 )
 
 //export run_proxy_server
diff --git a/python/util.py b/python/util.py
index 0ae68c1ed53766cb7f4f623e3a5f4fb50f7eb095..03442a7665a1fea5657b602d518bb5c21eda8281 100644
--- a/python/util.py
+++ b/python/util.py
@@ -31,6 +31,10 @@ def find_package(pkgname):
 
 
 def gen_pipeline_code(package_name):
+    ret = os.system("cd {}/pipeline/gateway/ && go mod init serving-gateway".
+                    format(package_name))
+    ret = os.system("cd {}/pipeline/gateway/ && go mod vendor && go mod tidy".
+                    format(package_name))
     # pipeline service proto
     protoc.main((
         '',
@@ -44,8 +48,8 @@ def gen_pipeline_code(package_name):
     ret = os.system(
         "cd {}/pipeline/gateway/proto/ && "
         "../../../../../third_party/install/protobuf/bin/protoc -I. "
-        "-I$GOPATH/src "
-        "-I$GOPATH/src/github.com/grpc-ecosystem/grpc-gateway/third_party/googleapis "
+        "-I$GOPATH/pkg/mod "
+        "-I$GOPATH/pkg/mod/github.com/grpc-ecosystem/grpc-gateway\@v1.15.2/third_party/googleapis "
         "--go_out=plugins=grpc:. "
         "gateway.proto".format(package_name))
     if ret != 0:
@@ -54,8 +58,8 @@ def gen_pipeline_code(package_name):
     ret = os.system(
         "cd {}/pipeline/gateway/proto/ && "
         "../../../../../third_party/install/protobuf/bin/protoc -I. "
-        "-I$GOPATH/src "
-        "-I$GOPATH/src/github.com/grpc-ecosystem/grpc-gateway/third_party/googleapis "
+        "-I$GOPATH/pkg/mod "
+        "-I$GOPATH/pkg/mod/github.com/grpc-ecosystem/grpc-gateway\@v1.15.2/third_party/googleapis "
         "--grpc-gateway_out=logtostderr=true:. "
         "gateway.proto".format(package_name))
     if ret != 0:
diff --git a/tools/serving_build.sh b/tools/serving_build.sh
index b3ccce1058f3baf9e3de94f22634f74c33d1c84a..52f4a85ca0456570b79750ad0faa955a6530e232 100644
--- a/tools/serving_build.sh
+++ b/tools/serving_build.sh
@@ -22,6 +22,9 @@ function init() {
     export GOPATH=$HOME/go
     export PATH=$PATH:$GOPATH/bin
 
+    go env -w GO111MODULE=on
+    go env -w GOPROXY=https://goproxy.cn,direct
+
     go get -u github.com/grpc-ecosystem/grpc-gateway/protoc-gen-grpc-gateway
     go get -u github.com/grpc-ecosystem/grpc-gateway/protoc-gen-swagger
     go get -u github.com/golang/protobuf/protoc-gen-go