diff --git a/codeflash/optimization/optimizer.py b/codeflash/optimization/optimizer.py
index 8a2095653..68d0a04a7 100644
--- a/codeflash/optimization/optimizer.py
+++ b/codeflash/optimization/optimizer.py
@@ -70,6 +70,7 @@ def __init__(self, args: Namespace) -> None:
         self.current_worktree: Path | None = None
         self.original_args_and_test_cfg: tuple[Namespace, TestConfig] | None = None
         self.patch_files: list[Path] = []
+        self.normalized_imports_cache: dict[Path, ValidCode] = {}
 
     def run_benchmarks(
         self, file_to_funcs_to_optimize: dict[Path, list[FunctionToOptimize]], num_optimizable_functions: int