diff --git a/.idea/runConfigurations/compare_lists.xml b/.idea/runConfigurations/compare_lists.xml
index 3240522a8..bccbe4d76 100644
--- a/.idea/runConfigurations/compare_lists.xml
+++ b/.idea/runConfigurations/compare_lists.xml
@@ -8,7 +8,7 @@
       <env name="PYTHONUNBUFFERED" value="1" />
     </envs>
     <option name="SDK_HOME" value="" />
-    <option name="SDK_NAME" value="$USER_HOME$/miniforge3/envs/codeflash311" />
+    <option name="SDK_NAME" value="$USER_HOME$/miniforge3/envs/codeflash312" />
     <option name="WORKING_DIRECTORY" value="$PROJECT_DIR$/cli" />
     <option name="IS_MODULE_SDK" value="false" />
     <option name="ADD_CONTENT_ROOTS" value="true" />
diff --git a/cli/codeflash/optimization/optimizer.py b/cli/codeflash/optimization/optimizer.py
index c2c302f20..7f7792f26 100644
--- a/cli/codeflash/optimization/optimizer.py
+++ b/cli/codeflash/optimization/optimizer.py
@@ -2,6 +2,7 @@ from __future__ import annotations
 
 import concurrent.futures
 import os
+import shutil
 import subprocess
 import tempfile
 import time
@@ -193,11 +194,8 @@ class Optimizer:
                             subprocess.run(["git", "worktree", "remove", "-f", worktree], check=True)
                         except subprocess.CalledProcessError as e:
                             logger.warning(f"Error deleting worktree: {e}")
-                    for root, dirs, files in worktree_root.walk(top_down=False):
-                        for name in files:
-                            (root / name).unlink()
-                        for name in dirs:
-                            (root / name).rmdir()
+                        shutil.rmtree(worktree)
+                    shutil.rmtree(worktree_root)
 
                     if is_successful(best_optimization):
                         optimizations_found += 1