diff --git a/ot/gromov.py b/ot/gromov.py
index 7de85988e..bc1c8e593 100644
--- a/ot/gromov.py
+++ b/ot/gromov.py
@@ -70,9 +70,12 @@ def init_matrix(C1, C2, p, q, loss_fun='square_loss'):
         Metric cost matrix in the source space
     C2 : array-like, shape (nt, nt)
         Metric cost matrix in the target space
-    T :  array-like, shape (ns, nt)
-        Coupling between source and target spaces
     p : array-like, shape (ns,)
+        Probability distribution in the source space
+    q : array-like, shape (nt,)
+        Probability distribution in the target space
+    loss_fun : str, optional
+        Name of loss function to use: either 'square_loss' or 'kl_loss' (default='square_loss')
 
     Returns
     -------