diff --git a/python/tests/test_preequilibration.py b/python/tests/test_preequilibration.py
index bf81e0256b..c7e254114e 100644
--- a/python/tests/test_preequilibration.py
+++ b/python/tests/test_preequilibration.py
@@ -18,7 +18,12 @@
 def preeq_fixture(pysb_example_presimulation_module):
     model = pysb_example_presimulation_module.getModel()
     model.setReinitializeFixedParameterInitialStates(True)
-
+    model.setSteadyStateComputationMode(
+        amici.SteadyStateComputationMode.integrateIfNewtonFails
+    )
+    model.setSteadyStateSensitivityMode(
+        amici.SteadyStateSensitivityMode.integrateIfNewtonFails
+    )
     solver = model.getSolver()
     solver.setSensitivityOrder(amici.SensitivityOrder.first)
     solver.setSensitivityMethod(amici.SensitivityMethod.forward)
diff --git a/python/tests/test_pregenerated_models.py b/python/tests/test_pregenerated_models.py
index 5a110cdfc2..e9daf13ba8 100755
--- a/python/tests/test_pregenerated_models.py
+++ b/python/tests/test_pregenerated_models.py
@@ -63,6 +63,13 @@ def test_pregenerated_model(sub_test, case):
     amici.readModelDataFromHDF5(
         options_file, model.get(), f"/{sub_test}/{case}/options"
     )
+    if model_name == "model_steadystate":
+        model.setSteadyStateComputationMode(
+            amici.SteadyStateComputationMode.integrateIfNewtonFails
+        )
+        model.setSteadyStateSensitivityMode(
+            amici.SteadyStateSensitivityMode.integrateIfNewtonFails
+        )
     amici.readSolverSettingsFromHDF5(
         options_file, solver.get(), f"/{sub_test}/{case}/options"
     )
diff --git a/python/tests/test_swig_interface.py b/python/tests/test_swig_interface.py
index f61b28ea4d..9686a25d94 100644
--- a/python/tests/test_swig_interface.py
+++ b/python/tests/test_swig_interface.py
@@ -105,12 +105,12 @@ def test_copy_constructors(pysb_example_presimulation_module):
     # `pysb_example_presimulation_module.getModel()`.
     "StateIsNonNegative": None,
     "SteadyStateComputationMode": [
-        2,
-        1,
+        amici.SteadyStateComputationMode.integrationOnly,
+        amici.SteadyStateComputationMode.integrateIfNewtonFails,
     ],
     "SteadyStateSensitivityMode": [
-        2,
-        1,
+        amici.SteadyStateSensitivityMode.integrationOnly,
+        amici.SteadyStateSensitivityMode.integrateIfNewtonFails,
     ],
     ("t0", "setT0"): [
         0.0,