diff --git a/src/arcaea_offline_ocr/device/__init__.py b/src/arcaea_offline_ocr/device/__init__.py
index e69de29..bd6cd44 100644
--- a/src/arcaea_offline_ocr/device/__init__.py
+++ b/src/arcaea_offline_ocr/device/__init__.py
@@ -0,0 +1,2 @@
+from .common import DeviceOcrResult
+from .ocr import DeviceOcr
diff --git a/src/arcaea_offline_ocr/device/rois/__init__.py b/src/arcaea_offline_ocr/device/rois/__init__.py
new file mode 100644
index 0000000..e73f32a
--- /dev/null
+++ b/src/arcaea_offline_ocr/device/rois/__init__.py
@@ -0,0 +1,3 @@
+from .definition import *
+from .extractor import *
+from .masker import *
diff --git a/src/arcaea_offline_ocr/device/rois/definition/__init__.py b/src/arcaea_offline_ocr/device/rois/definition/__init__.py
index 4083875..37f6ef1 100644
--- a/src/arcaea_offline_ocr/device/rois/definition/__init__.py
+++ b/src/arcaea_offline_ocr/device/rois/definition/__init__.py
@@ -1 +1,2 @@
 from .auto import *
+from .common import DeviceRois
diff --git a/src/arcaea_offline_ocr/device/rois/extractor/common.py b/src/arcaea_offline_ocr/device/rois/extractor/common.py
index e672e8b..a90a0a4 100644
--- a/src/arcaea_offline_ocr/device/rois/extractor/common.py
+++ b/src/arcaea_offline_ocr/device/rois/extractor/common.py
@@ -5,9 +5,9 @@ from ..definition.common import DeviceRois
 
 
 class DeviceRoisExtractor:
-    def __init__(self, img: cv2.Mat, sizes: DeviceRois):
+    def __init__(self, img: cv2.Mat, rois: DeviceRois):
         self.img = img
-        self.sizes = sizes
+        self.sizes = rois
 
     def __construct_int_rect(self, rect):
         return tuple(round(r) for r in rect)