diff --git a/src/compute_state.cc b/src/compute_state.cc
index fbbc3a8e598db8432a2fde504b32161989840403..94e1a0d7156b1772de7015c173add3ca87fb3f86 100644
--- a/src/compute_state.cc
+++ b/src/compute_state.cc
@@ -41,8 +41,7 @@ double compute_state_update_bisection(double h, double unorm, double L,
 double compute_state_update(double h, double unorm, double L,
                             double old_state) {
   double ret1 = compute_state_update_bisection(h, unorm, L, old_state);
-  assert(std::abs(1.0 / h * ret1 - (old_state - unorm / L) / h -
-                  std::exp(-ret1)) < 1e-8);
+  assert(std::abs(ret1 - old_state + unorm / L - h * std::exp(-ret1)) < 1e-10);
 
   return ret1;
 }