diff --git a/tests/e2e/conftest.py b/tests/e2e/conftest.py
index a949d908..aadbb1d8 100644
--- a/tests/e2e/conftest.py
+++ b/tests/e2e/conftest.py
@@ -1,4 +1,5 @@
 import json
+import os
 import shutil
 import sys
 import tempfile
@@ -30,7 +31,8 @@ def working_directory() -> str:
         def _assert_working_directory_empty(self, is_error: bool = False):
             files = [str(file_path) for file_path in _get_files_in_directory(temp_dir)]
             num_files = len(files)
-            shutil.rmtree(temp_dir)
+            if os.path.isdir(temp_dir):
+                shutil.rmtree(temp_dir)
 
             if not is_error:
                 if num_files > 0: