diff --git a/lib/crewai/src/crewai/utilities/tool_files.py b/lib/crewai/src/crewai/utilities/tool_files.py
index 622fda9b8..0049e9a4c 100644
--- a/lib/crewai/src/crewai/utilities/tool_files.py
+++ b/lib/crewai/src/crewai/utilities/tool_files.py
@@ -82,7 +82,7 @@ def extract_files_from_tool_result(
         A ``(files, message)`` tuple. ``files`` is ``None`` when no files
         were detected.
     """
-    files: dict[str, BaseFile] = {}
+    files: dict[str, FileInput] = {}
 
     if _is_base_file(result):
         key = _file_key(result, used=set(), default_index=0)