diff --git a/l2gv2/anomaly_detection.py b/l2gv2/anomaly_detection.py
index ef37111..779cd27 100644
--- a/l2gv2/anomaly_detection.py
+++ b/l2gv2/anomaly_detection.py
@@ -5,7 +5,9 @@
 from .patch.patch import Patch
 
 
-def raw_anomaly_score_node_patch(aligned_patch_emb, emb, node) -> np.floating[Any]:
+def raw_anomaly_score_node_patch(
+    aligned_patch_emb: Patch, emb: list[Patch], node: int
+) -> np.floating[Any]:
     """TODO: docstring for `raw_anomaly_score_node_patch`
 
     Args: