1 files changed, 2 insertions, 3 deletions
diff --git a/src/qms.c b/src/qms.c
index b5abcd2..62d8da9 100644
--- a/src/qms.c
+++ b/src/qms.c
@@ -1391,15 +1391,14 @@ skip_solve:
 
         const double vel_fact = 1.0 / (1 << (MIN(reflevel, ms->solve_level) - reflevel_top));
 
-        qms_assert(fabs(time - W_pred1_time[reflevel]) < 1e-13);
-
 #pragma omp parallel for
         for (int i = 0; i < grid_size; i++) {
             const double sol_new = W_val[i];
-            const double delta   = sol_new - W_pred1[i];
+            const double delta   = sol_new - W_eval[i];
 
             W_val0[i] = W_val1[i] + delta - delta * vel_fact;
             W_val1[i] = sol_new;
+            W_eval[i] = 2 * W_val1[i] - W_val0[i];
         }
         W_val0_time[reflevel] = W_val1_time[reflevel];
         W_val1_time[reflevel] = time;