diff --git a/tests/unit/test_2035_coverage.py b/tests/unit/test_2035_coverage.py
index b8bbedf6d..7c682b876 100644
--- a/tests/unit/test_2035_coverage.py
+++ b/tests/unit/test_2035_coverage.py
@@ -87,7 +87,9 @@ _GEN_TASKS_PATH = Path(__file__).resolve().parents[2] / "apps/api/app/api/routes
 def _load_infer_func():
     src = _GEN_TASKS_PATH.read_text()
     start = src.index("# #2035：文案关键词")
-    end = src.index("from packages.middleware")
+    # 用紧跟 _infer_expected_categories 后的 logger 行作为结束锚点
+    end_marker = "\nlogger = logging.getLogger"
+    end = src.index(end_marker, start)
     code = src[start:end]
     ns: dict = {}
     exec(code, ns)