diff --git a/src/caf/distribute/gravity_model/multi_area.py b/src/caf/distribute/gravity_model/multi_area.py
index ccd4d17..36ff298 100644
--- a/src/caf/distribute/gravity_model/multi_area.py
+++ b/src/caf/distribute/gravity_model/multi_area.py
@@ -501,7 +501,6 @@ def __init__(
         col_targets: np.ndarray,
         cost_matrix: np.ndarray,
         cost_function: cost_functions.CostFunction,
-        # TODO(kf) move these parameters as inputs of calibrate and run
     ):
         super().__init__(cost_function=cost_function, cost_matrix=cost_matrix)