diff --git a/salve/wrappers.py b/salve/wrappers.py
index a8763c2..d79f6d3 100644
--- a/salve/wrappers.py
+++ b/salve/wrappers.py
@@ -1,4 +1,4 @@
-from collegamento import Request
+from collegamento import Request, FileServer
 from pyeditorconfig import get_config
 from token_tools import Token, normal_text_range
 
@@ -11,7 +11,7 @@
 )
 
 
-def find_autocompletions_request_wrapper(request: Request) -> list[str]:
+def find_autocompletions_request_wrapper(server: FileServer, request: Request) -> list[str]:
     return find_autocompletions(
         full_text=request["file"],  # type: ignore
         expected_keywords=request["expected_keywords"],  # type: ignore
@@ -19,7 +19,7 @@ def find_autocompletions_request_wrapper(request: Request) -> list[str]:
     )
 
 
-def get_replacements_request_wrapper(request: Request) -> list[str]:
+def get_replacements_request_wrapper(server: FileServer, request: Request) -> list[str]:
     return get_replacements(
         full_text=request["file"],  # type: ignore
         expected_keywords=request["expected_keywords"],  # type: ignore
@@ -27,7 +27,7 @@ def get_replacements_request_wrapper(request: Request) -> list[str]:
     )
 
 
-def get_highlights_request_wrapper(request: Request) -> list[Token]:
+def get_highlights_request_wrapper(server: FileServer, request: Request) -> list[Token]:
     return get_highlights(
         full_text=request["file"],  # type: ignore
         language=request["language"],  # type: ignore
@@ -35,11 +35,11 @@ def get_highlights_request_wrapper(request: Request) -> list[Token]:
     )
 
 
-def editorconfig_request_wrapper(request: Request) -> dict:
+def editorconfig_request_wrapper(server: FileServer, request: Request) -> dict:
     return get_config(request["file_path"])  # type: ignore
 
 
-def get_definition_request_wrapper(request: Request) -> Token:
+def get_definition_request_wrapper(server: FileServer, request: Request) -> Token:
     return get_definition(
         request["file"],  # type: ignore
         request["definition_starters"],  # type: ignore
@@ -47,7 +47,7 @@ def get_definition_request_wrapper(request: Request) -> Token:
     )
 
 
-def get_special_tokens_request_wrapper(request: Request) -> list[Token]:
+def get_special_tokens_request_wrapper(server: FileServer, request: Request) -> list[Token]:
     return get_special_tokens(
         request["file"],  # type: ignore
         normal_text_range(request["file"], request["text_range"])[  # type: ignore