diff --git a/setup.py b/setup.py
index c564b3d..2e8007f 100644
--- a/setup.py
+++ b/setup.py
@@ -78,6 +78,7 @@ def run(self):
           '--proto_path={}'.format(grpc_protos_include),
           '--proto_path={}'.format(_ROOT_DIR),
           '--python_out={}'.format(_ROOT_DIR),
+          '--pyi_out={}'.format(_ROOT_DIR),
           '--grpc_python_out={}'.format(_ROOT_DIR),
           os.path.join(_ROOT_DIR, proto_path),
       ]