diff --git a/src/explainers/aggregators/self_influence.py b/src/explainers/aggregators/self_influence.py
index b0f83a14..e7be7367 100644
--- a/src/explainers/aggregators/self_influence.py
+++ b/src/explainers/aggregators/self_influence.py
@@ -29,3 +29,4 @@ def get_self_influence_ranking(
             train_ids=[i],
             **explain_fn_kwargs,
         )
+    return self_inf.argsort()