diff --git a/pymilvus/grpc_gen/python_gen.sh b/pymilvus/grpc_gen/python_gen.sh
index 6bff7856a..b2d3dfadb 100755
--- a/pymilvus/grpc_gen/python_gen.sh
+++ b/pymilvus/grpc_gen/python_gen.sh
@@ -3,6 +3,8 @@
 OUTDIR=.
 PROTO_DIR="milvus-proto/proto"
 
+python -m pip install "grpcio-tools==$(python3 -c 'import grpc; print(grpc.__version__)')"
+
 python -m grpc_tools.protoc -I ${PROTO_DIR} --python_out=${OUTDIR} --pyi_out=${OUTDIR} ${PROTO_DIR}/common.proto
 python -m grpc_tools.protoc -I ${PROTO_DIR} --python_out=${OUTDIR} --pyi_out=${OUTDIR} ${PROTO_DIR}/schema.proto
 python -m grpc_tools.protoc -I ${PROTO_DIR} --python_out=${OUTDIR} --pyi_out=${OUTDIR} --grpc_python_out=${OUTDIR}  ${PROTO_DIR}/milvus.proto
diff --git a/setup.py b/setup.py
index b06c1f96b..8001038b0 100644
--- a/setup.py
+++ b/setup.py
@@ -20,7 +20,6 @@
     include_package_data=True,
     install_requires=[
         "grpcio>=1.49.1,<=1.53.0",
-        "grpcio-tools>=1.49.1,<=1.53.0",
         "mmh3>=2.0",
         "ujson>=2.0.0",
         "pandas>=1.2.4",