diff --git a/tests/problems/base/test_general_problem.py b/tests/problems/base/test_general_problem.py
index 869b0ea7f..dc05b09b9 100644
--- a/tests/problems/base/test_general_problem.py
+++ b/tests/problems/base/test_general_problem.py
@@ -288,7 +288,7 @@ def test_set_graph_x_y(self, adata_x: AnnData, adata_y: AnnData, ts: Tuple[Optio
         assert ta2.tag == Tag.GRAPH
         assert ta2.cost == "geodesic"
 
-        prob1 = prob1.solve(lse_mode=False, epsilon=10.0)
+        prob1 = prob1.solve(epsilon=10.0)
 
         prob2 = OTProblem(adata_x, adata_y)
         prob2 = prob2.prepare(