diff --git a/extraction/core/functions/loaders.py b/extraction/core/functions/loaders.py
index 46e8ec89834346d257cd0eaf9a724258b6bc41c4..cb26d47ea832a6365bd655d0b3e29e3cc73114cf 100644
--- a/extraction/core/functions/loaders.py
+++ b/extraction/core/functions/loaders.py
@@ -1,3 +1,4 @@
+import typing as t
 from inspect import getfullargspec, getmembers, isfunction
 
 import numpy as np
@@ -12,6 +13,7 @@ Load functions for analysing cells and their background.
 Note that inspect.getmembers returns a list of function names and functions, and inspect.getfullargspec returns a function's arguments.
 """
 
+
 def load_cellfuns_core():
     """
     Load functions from the cell module and return as a dict.
@@ -24,7 +26,9 @@ def load_cellfuns_core():
     }
 
 
-def load_custom_args():
+def load_custom_args() -> t.Tuple[
+    (t.Dict[str, t.Callable], t.Dict[str, t.List[str]])
+]:
     """
     Load custom functions from the localisation module and return the functions and any additional arguments, other than cell_mask and trap_image, as dictionaries.
     """