From 41c7d0cab38820ede56666e063ea4d2eecc342f7 Mon Sep 17 00:00:00 2001
From: lahramon <41509708+lahramon@users.noreply.github.com>
Date: Mon, 20 Nov 2023 10:28:52 +0100
Subject: [PATCH] Switch updating time stamp to update with time stamp after
 predicting (#678)

---
 probdiffeq/solvers/strategies/strategy.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/probdiffeq/solvers/strategies/strategy.py b/probdiffeq/solvers/strategies/strategy.py
index f911e0cb..0922a4da 100644
--- a/probdiffeq/solvers/strategies/strategy.py
+++ b/probdiffeq/solvers/strategies/strategy.py
@@ -100,10 +100,10 @@ def init(self, t, posterior, /) -> _State:
     def predict_error(self, state: _State, /, *, dt, vector_field):
         """Predict the error of an upcoming step."""
         hidden, extra = self.extrapolation.begin(state.hidden, state.aux_extra, dt=dt)
+        t = state.t + dt
         error, observed, corr = self.correction.estimate_error(
-            hidden, state.aux_corr, vector_field=vector_field, t=state.t
+            hidden, state.aux_corr, vector_field=vector_field, t=t
         )
-        t = state.t + dt
         state = _State(t=t, hidden=hidden, aux_extra=extra, aux_corr=corr)
         return error, observed, state