diff --git a/chapter2/hyperelasticity.ipynb b/chapter2/hyperelasticity.ipynb
index 1271a4bb..818d3549 100644
--- a/chapter2/hyperelasticity.ipynb
+++ b/chapter2/hyperelasticity.ipynb
@@ -437,7 +437,7 @@
     "for n in range(1, 10):\n",
     "    T.value[2] = n * tval0\n",
     "    problem.solve()\n",
-    "    converged = problem.solver.getConvergedReason() > 0\n",
+    "    converged = problem.solver.getConvergedReason()\n",
     "    num_its = problem.solver.getIterationNumber()\n",
     "    assert converged > 0, f\"Solver did not converge with reason {converged}.\"\n",
     "\n",
diff --git a/chapter2/hyperelasticity.py b/chapter2/hyperelasticity.py
index a088f592..b3fb1069 100644
--- a/chapter2/hyperelasticity.py
+++ b/chapter2/hyperelasticity.py
@@ -209,7 +209,7 @@ def right(x):
 for n in range(1, 10):
     T.value[2] = n * tval0
     problem.solve()
-    converged = problem.solver.getConvergedReason() > 0
+    converged = problem.solver.getConvergedReason()
     num_its = problem.solver.getIterationNumber()
     assert converged > 0, f"Solver did not converge with reason {converged}."