diff --git a/apps/ai/helmet/algo.py b/apps/ai/helmet/algo.py
index a7a8c8eb..0b4f92d2 100644
--- a/apps/ai/helmet/algo.py
+++ b/apps/ai/helmet/algo.py
@@ -4,8 +4,8 @@ import base64
 from io import BytesIO
 import requests
 import grpc
-from . import BeiHangGrpc_pb2_grpc
-from . import BeiHangGrpc_pb2
+from apps.ai.helmet import BeiHangGrpc_pb2_grpc
+from apps.ai.helmet import BeiHangGrpc_pb2
 # from apps.utils.tools import convert_to_base64