diff --git a/tests/problems/base/test_general_problem.py b/tests/problems/base/test_general_problem.py
index dc05b09b..094eb61e 100644
--- a/tests/problems/base/test_general_problem.py
+++ b/tests/problems/base/test_general_problem.py
@@ -313,7 +313,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"
 
-        prob2 = prob2.solve(lse_mode=False, epsilon=10.0)
+        prob2 = prob2.solve(epsilon=10.0)
 
         assert not np.allclose(prob1.solution._output.geom.cost_matrix, prob2.solution._output.geom.cost_matrix)