diff --git a/tests/e2e/cost_calculation/generate_expected.py b/tests/e2e/cost_calculation/generate_expected.py
index c093ecbe0ea..e243e477839 100644
--- a/tests/e2e/cost_calculation/generate_expected.py
+++ b/tests/e2e/cost_calculation/generate_expected.py
@@ -16,14 +16,10 @@ import json
 import sys
 from collections.abc import Mapping
 from dataclasses import dataclass
-from pathlib import Path
 from types import MappingProxyType
 from typing import Final
 
-sys.path.insert(0, str(Path(__file__).resolve().parent))
-sys.path.insert(0, str(Path(__file__).resolve().parent.parent))
-
-from cost_matrix import (  # noqa: E402  # path bootstrap before package-local imports
+from cost_matrix import (
     EXPECTED_PATH,
     FRONTIER_MODELS,
     TIER_THRESHOLD_TOKENS,