diff --git a/blendsql/ingredients/builtin/map/main.py b/blendsql/ingredients/builtin/map/main.py
index 5340b7c7..9344d4a0 100644
--- a/blendsql/ingredients/builtin/map/main.py
+++ b/blendsql/ingredients/builtin/map/main.py
@@ -139,7 +139,7 @@ def run(
         value_limit: Union[int, None] = None,
         example_outputs: Optional[str] = None,
         output_type: Optional[str] = None,
-        regex: Optional[str] = None,
+        regex: Optional[Callable[[int], str]] = None,
         table_to_title: Optional[Dict[str, str]] = None,
         **kwargs,
     ) -> Iterable[Any]: