diff --git a/.github/workflows/codeflash.yaml b/.github/workflows/codeflash.yaml
index 557e27c5..f450e547 100644
--- a/.github/workflows/codeflash.yaml
+++ b/.github/workflows/codeflash.yaml
@@ -36,8 +36,13 @@ jobs:
           pip install poetry
           poetry install --all-extras
           poetry add codeflash
-
+      - name: create test dir
+        run: |
+          mkdir -p codeflash-tests
       - name: Run Codeflash to optimize code
         run: |
           poetry env use python
           poetry run codeflash
+      - name: remove test dir
+        run: |
+          rm -rf codeflash-tests
\ No newline at end of file
diff --git a/pyproject.toml b/pyproject.toml
index 3b51fd03..0e583b38 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -139,7 +139,7 @@ skyvern = "skyvern.cli.commands:app"
 [tool.codeflash]                                                                                                                                                                                                                                                                                                             
 # All paths are relative to this pyproject.toml's directory.                                                                                                                                                                                                                                                                 
 module-root = "skyvern"                                                                                                                                                                                                                                                                                                      
-tests-root = "tests"                                                                                                                                                                                                                                                                                                         
+tests-root = "codeflash-tests"                                                                                                                                                                                                                                                                                                         
 test-framework = "pytest"                                                                                                                                                                                                                                                                                                    
 ignore-paths = ["skyvern/client"]                                                                                                                                                                                                                                                                                                         
 formatter-cmds = ["ruff check --exit-zero --fix $file", "ruff format $file"] 
\ No newline at end of file