From 40068224f31cbd6318c3ca9304eb9394380192a2 Mon Sep 17 00:00:00 2001
From: jubicker <113909589+jubicker@users.noreply.github.com>
Date: Thu, 16 Oct 2025 16:11:03 +0200
Subject: [PATCH] update internal waiting time at the end of advance

---
 cpp/models/smm/simulation.h | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/cpp/models/smm/simulation.h b/cpp/models/smm/simulation.h
index 29c4aeda24..49d4dcbb8f 100644
--- a/cpp/models/smm/simulation.h
+++ b/cpp/models/smm/simulation.h
@@ -127,6 +127,10 @@ class Simulation
         if (last_result_time < tmax) {
             m_result.add_time_point(tmax);
             m_result.get_last_value() = m_result[m_result.get_num_time_points() - 2];
+            // update internal times
+            for (size_t i = 0; i < m_internal_time.size(); i++) {
+                m_internal_time[i] += m_current_rates[i] * (tmax - current_time);
+            }
         }
         return m_result.get_last_value();
     }