diff --git a/Makefile b/Makefile
index 36b301382..a15a6ade7 100644
--- a/Makefile
+++ b/Makefile
@@ -17,8 +17,6 @@ OSX_APP_GUI ?= webview
 
 FRONTEND_EXPORT_DIR = frontend/out
 
-PYTHON_PACKAGE_DIR = $(PWD)/python/bbgo
-
 all: bbgo-linux bbgo-darwin
 
 $(BIN_DIR):
@@ -257,11 +255,9 @@ install-grpc-tools:
 # replace `import bbgo_pb2` by `from . import bbgo_pb2` to use relative import
 grpc-py:
 	python -m grpc_tools.protoc -I$(PWD)/pkg/pb \
-		--python_out=$(PYTHON_PACKAGE_DIR) \
-		--grpc_python_out=$(PYTHON_PACKAGE_DIR) \
+		--python_out=$(PWD)/python \
+		--grpc_python_out=$(PWD)/python \
 		$(PWD)/pkg/pb/bbgo.proto
-	cat $(PYTHON_PACKAGE_DIR)/bbgo_pb2_grpc.py | sed -e 's/import\ bbgo_pb2/from\ .\ import\ bbgo_pb2/g' > tmp.py
-	mv tmp.py $(PYTHON_PACKAGE_DIR)/bbgo_pb2_grpc.py
 
 clean:
 	rm -rf $(BUILD_DIR) $(DIST_DIR) $(FRONTEND_EXPORT_DIR) $(GRPC_GO_DEPS) pkg/pb/*.pb.go
diff --git a/python/bbgo/services.py b/python/bbgo/services.py
index 989339a31..fe40a233d 100644
--- a/python/bbgo/services.py
+++ b/python/bbgo/services.py
@@ -1,7 +1,7 @@
 from typing import List
 
-from . import bbgo_pb2
-from . import bbgo_pb2_grpc
+import bbgo_pb2
+import bbgo_pb2_grpc
 
 
 class UserDataService(object):
diff --git a/python/bbgo/stream.py b/python/bbgo/stream.py
index 99bf37229..01979bd3e 100644
--- a/python/bbgo/stream.py
+++ b/python/bbgo/stream.py
@@ -2,11 +2,10 @@ import asyncio
 from typing import Callable
 from typing import List
 
+import bbgo_pb2
+import bbgo_pb2_grpc
 import grpc
 
-from . import bbgo_pb2
-from . import bbgo_pb2_grpc
-
 
 class Stream(object):
 
diff --git a/python/bbgo/bbgo_pb2.py b/python/bbgo_pb2.py
similarity index 100%
rename from python/bbgo/bbgo_pb2.py
rename to python/bbgo_pb2.py
diff --git a/python/bbgo/bbgo_pb2_grpc.py b/python/bbgo_pb2_grpc.py
similarity index 99%
rename from python/bbgo/bbgo_pb2_grpc.py
rename to python/bbgo_pb2_grpc.py
index 54580eeff..9e3ac27ae 100644
--- a/python/bbgo/bbgo_pb2_grpc.py
+++ b/python/bbgo_pb2_grpc.py
@@ -2,7 +2,7 @@
 """Client and server classes corresponding to protobuf-defined services."""
 import grpc
 
-from . import bbgo_pb2 as bbgo__pb2
+import bbgo_pb2 as bbgo__pb2
 
 
 class MarketDataServiceStub(object):
diff --git a/python/pyproject.toml b/python/pyproject.toml
index 5f6a5b14b..df2941183 100644
--- a/python/pyproject.toml
+++ b/python/pyproject.toml
@@ -3,6 +3,11 @@ name = "bbgo"
 version = "0.1.0"
 description = ""
 authors = ["なるみ <weaper@gmail.com>"]
+packages = [
+    { include = "bbgo" },
+    { include = "bbgo_pb2.py" },
+    { include = "bbgo_pb2_grpc.py" },
+]
 
 [tool.poetry.dependencies]
 python = "^3.8"
diff --git a/python/tests/servicer.py b/python/tests/servicer.py
index 5ae76873f..5a367663c 100644
--- a/python/tests/servicer.py
+++ b/python/tests/servicer.py
@@ -1,5 +1,5 @@
-from bbgo import bbgo_pb2
-from bbgo import bbgo_pb2_grpc
+import bbgo_pb2
+import bbgo_pb2_grpc
 
 
 class TestTradingServicer(bbgo_pb2_grpc.TradingServiceServicer):
diff --git a/python/tests/test_grpc.py b/python/tests/test_grpc.py
index b98d2a174..99a843f34 100644
--- a/python/tests/test_grpc.py
+++ b/python/tests/test_grpc.py
@@ -1,11 +1,12 @@
 from concurrent import futures
 
+import bbgo_pb2
+import bbgo_pb2_grpc
 import grpc
 import pytest
 
-from bbgo import MarketService, TradingService
-from bbgo import bbgo_pb2
-from bbgo import bbgo_pb2_grpc
+from bbgo import MarketService
+from bbgo import TradingService
 from tests.servicer import TestTradingServicer