diff --git a/tests/test_yuanbao_pipeline.py b/tests/test_yuanbao_pipeline.py
index 3d5321bdc75..023d949b2cc 100644
--- a/tests/test_yuanbao_pipeline.py
+++ b/tests/test_yuanbao_pipeline.py
@@ -20,7 +20,7 @@ if _REPO_ROOT not in sys.path:
     sys.path.insert(0, _REPO_ROOT)
 
 import pytest
-from unittest.mock import AsyncMock, MagicMock
+from unittest.mock import AsyncMock, MagicMock, patch
 
 from gateway.platforms.yuanbao import (
     InboundContext,