diff --git a/src/graphnet/training/weight_fitting.py b/src/graphnet/training/weight_fitting.py
index 2e4a1879d..e66c2d4c5 100644
--- a/src/graphnet/training/weight_fitting.py
+++ b/src/graphnet/training/weight_fitting.py
@@ -78,10 +78,10 @@ def fit(
                 happen in this space.
             db_count_norm: If given, the total sum of the weights for the given
                 db will be this number.
-            automatic_log_bins: If True, the bins are generated as a log10 space
-                between the min and max of the variable.
-            max_weight: If given, the weights are capped such that the sum of a
-                single event's weight cannot exceed this number times the sum of
+            automatic_log_bins: If True, the bins are generated as a log10
+                space between the min and max of the variable.
+            max_weight: If given, the weights are capped such that a single
+                event weight cannot exceed this number times the sum of
                 all weights.
             **kwargs: Additional arguments passed to `_fit_weights`.