diff --git a/server/build.sh b/server/build.sh
index 0d7ba99df46bd47ac1d44a54d1fb62a594c54de8..c9c490768e44e3faa525f9f58ff68e1ff9d4f9a3 100644
--- a/server/build.sh
+++ b/server/build.sh
@@ -1,2 +1,7 @@
-protoc visualdl/onnx/onnx.proto --python_out .
-python setup.py bdist_wheel
+script=$(readlink -f "$0")
+script_path=$(dirname "$script")
+
+pushd $script_path
+  protoc visualdl/onnx/onnx.proto --python_out .
+  python setup.py bdist_wheel
+popd
diff --git a/server/setup.py b/server/setup.py
index 6b8c71ed23929662725d683af708a20e10b5eb10..b20a62b86891819e50e76bcb561d6ac4f4f0e829 100644
--- a/server/setup.py
+++ b/server/setup.py
@@ -1,6 +1,9 @@
 from setuptools import setup
 
-packages = ['visualdl', 'visualdl.frontend.dist', 'visualdl.mock']
+packages = ['visualdl',
+            'visualdl.onnx',
+            'visualdl.mock',
+            'visualdl.frontend.dist']
 
 setup(
     name="visualdl",