diff --git a/propulate/surrogate.py b/propulate/surrogate.py
index 54bf468..aed59d0 100644
--- a/propulate/surrogate.py
+++ b/propulate/surrogate.py
@@ -291,6 +291,7 @@ def merge(self, data: np.ndarray) -> None:
             self.first_run = False
             return
 
+        # TODO test if this is the right way around
         # Merged run's final loss is better than baseline.
         if self.baseline[-1] < data[-1]:
             self.baseline = data.copy()
diff --git a/tests/test_multi_rank_workers.py b/tests/test_multi_rank_workers.py
index c2ffa53..ebe6c60 100644
--- a/tests/test_multi_rank_workers.py
+++ b/tests/test_multi_rank_workers.py
@@ -84,5 +84,7 @@ def test_multi_rank_workers(mpi_tmp_path: pathlib.Path) -> None:
         logging_interval=10,  # Logging interval
         debug=1,  # Debug level
     )
-    islands.summarize(top_n=1, debug=1)
     log.handlers.clear()
+
+
+# TODO surrogate