diff --git a/codeflash/optimization/optimizer.py b/codeflash/optimization/optimizer.py
index cd53cc56c..f794fc01b 100644
--- a/codeflash/optimization/optimizer.py
+++ b/codeflash/optimization/optimizer.py
@@ -312,6 +312,8 @@ def run(self) -> None:
                             original_module_ast=original_module_ast,
                             original_module_path=original_module_path,
                         )
+                        if function_optimizer is None:
+                            continue
 
                         self.current_function_optimizer = (
                             function_optimizer  # needed to clean up from the outside of this function