diff --git a/apps/api/app/api/routes/scripts_ai.py b/apps/api/app/api/routes/scripts_ai.py
index 78a999942..426936ea6 100644
--- a/apps/api/app/api/routes/scripts_ai.py
+++ b/apps/api/app/api/routes/scripts_ai.py
@@ -30,6 +30,7 @@ from app.services.script_asr_service import (
 from fastapi import APIRouter, Depends, HTTPException, status
 from sqlalchemy.orm import Session
 
+from app.dependencies import get_db_session
 from packages.middleware.points_gate import points_gate
 from packages.shared.ai_client import get_doubao_client
 
diff --git a/tests/unit/test_scripts_ai.py b/tests/unit/test_scripts_ai.py
index 87b9f2417..bb5a0c43e 100644
--- a/tests/unit/test_scripts_ai.py
+++ b/tests/unit/test_scripts_ai.py
@@ -13,6 +13,8 @@ from __future__ import annotations
 import sys
 from unittest.mock import MagicMock, patch
 
+import packages.middleware.points_gate as _pg_module
+
 import pydantic
 import pytest