diff --git a/demo/draw_bbox.py b/magic_pdf/libs/draw_bbox.py
similarity index 100%
rename from demo/draw_bbox.py
rename to magic_pdf/libs/draw_bbox.py
diff --git a/magic_pdf/pdf_parse_by_ocr.py b/magic_pdf/pdf_parse_by_ocr.py
index 81c5d970..b17a00d0 100644
--- a/magic_pdf/pdf_parse_by_ocr.py
+++ b/magic_pdf/pdf_parse_by_ocr.py
@@ -4,7 +4,7 @@ import time
 
 from loguru import logger
 
-from demo.draw_bbox import draw_layout_bbox, draw_text_bbox
+from magic_pdf.libs.draw_bbox import draw_layout_bbox, draw_text_bbox
 from magic_pdf.libs.commons import (
     read_file,
     join_path,