diff --git a/mle/agents/advisor.py b/mle/agents/advisor.py
index 2c96aa2..1a0fb9f 100644
--- a/mle/agents/advisor.py
+++ b/mle/agents/advisor.py
@@ -126,7 +126,8 @@ def __init__(self, model, console=None, mode='normal'):
             schema_web_search,
             schema_search_arxiv,
             schema_search_papers_with_code,
-            schema_preview_csv_data
+            schema_preview_csv_data,
+            schema_list_files
         ]
         if config_data.get('search_key'):
             self.functions.append(schema_web_search)