diff --git a/quantile_forest/_quantile_forest.py b/quantile_forest/_quantile_forest.py
index e2a80e7..5f05e53 100755
--- a/quantile_forest/_quantile_forest.py
+++ b/quantile_forest/_quantile_forest.py
@@ -934,8 +934,8 @@ def proximity_counts(
 
         Returns
         -------
-        proximities : list of tuples
-            List of tuples mapping sample indices to proximity counts.
+        proximities : list of tuples, length = n_samples
+            List of tuples mapping training sample indices to proximity counts.
 
         Notes
         -----