diff --git a/video_decoder.py b/video_decoder.py
index bab47ad..2bcc35f 100644
--- a/video_decoder.py
+++ b/video_decoder.py
@@ -3,7 +3,7 @@ import av
 import numpy as np
 from towhee.operator.base import PyOperator
 
-from cpu_decode import PyAVDecode
+from .cpu_decode import PyAVDecode
 
 logger = logging.getLogger()