diff --git a/api/extensions/ext_storage.py b/api/extensions/ext_storage.py
index db5a6e4812..e7bb907fd0 100644
--- a/api/extensions/ext_storage.py
+++ b/api/extensions/ext_storage.py
@@ -110,10 +110,10 @@ class Storage:
     def load_stream(self, filename: str) -> Generator:
         return self.storage_runner.load_stream(filename)
 
-    def download(self, filename, target_filepath):
+    def download(self, filename: str, target_filepath):
         self.storage_runner.download(filename, target_filepath)
 
-    def exists(self, filename):
+    def exists(self, filename: str):
         return self.storage_runner.exists(filename)
 
     def delete(self, filename: str):
diff --git a/api/pyrefly.toml b/api/pyrefly.toml
index 80ffba019d..01f4c5a529 100644
--- a/api/pyrefly.toml
+++ b/api/pyrefly.toml
@@ -1,9 +1,7 @@
 project-includes = ["."]
 project-excludes = [
-    "tests/",
     ".venv",
     "migrations/",
-    "core/rag",
 ]
 python-platform = "linux"
 python-version = "3.11.0"