diff --git a/src/codegen/extensions/tools/list_directory.py b/src/codegen/extensions/tools/list_directory.py
index b2668c435..357f303ca 100644
--- a/src/codegen/extensions/tools/list_directory.py
+++ b/src/codegen/extensions/tools/list_directory.py
@@ -140,9 +140,8 @@ def get_directory_info(dir_obj: Directory, current_depth: int) -> DirectoryInfo:
         """Helper function to get directory info recursively."""
         # Get direct files (always include files unless at max depth)
         all_files = []
-        for file in dir_obj.files(recursive=True):
-            if file.directory == dir_obj:
-                all_files.append(file.filepath.split("/")[-1])
+        for file_name in dir_obj.file_names:
+            all_files.append(file_name)
 
         # Get direct subdirectories
         subdirs = []
diff --git a/src/codegen/sdk/core/directory.py b/src/codegen/sdk/core/directory.py
index 504d608e0..806e90ff8 100644
--- a/src/codegen/sdk/core/directory.py
+++ b/src/codegen/sdk/core/directory.py
@@ -196,6 +196,11 @@ def item_names(self, recursive: bool = False) -> list[str]:
         """
         return self._files + self._subdirectories
 
+    @property
+    def file_names(self) -> list[str]:
+        """Get a list of all file names in the directory."""
+        return self._files
+
     @property
     def tree(self) -> list[Self | TFile]:
         """Get a recursive list of all files and subdirectories in the directory.