diff --git a/tests/unit/test_wasserstein.py b/tests/unit/test_wasserstein.py
index b69c428d..d73a41b2 100644
--- a/tests/unit/test_wasserstein.py
+++ b/tests/unit/test_wasserstein.py
@@ -123,7 +123,7 @@
 amg_options = {
     "linear_solver": "amg",
     "linear_solver_options": {
-        "tol": 1e-8,
+        "atol": 1e-8,
     },
     "formulation": "pressure",
 }
@@ -137,7 +137,7 @@
 ksp_krylov_options = {
     "linear_solver": "ksp",
     "linear_solver_options": {
-        "tol": 1e-8,
+        "rtol": 1e-8,
         "approach": "cg",
         "pc_type": "hypre",
     },
@@ -147,7 +147,7 @@
 ksp_block_krylov_options = {
     "linear_solver": "ksp",
     "linear_solver_options": {
-        "tol": 1e-8,
+        "rtol": 1e-8,
         "approach": "gmres",
         "pc_type": "hypre",
     },