From 979279b31c6df475eb612029ed7e4f7a0a5a8ac9 Mon Sep 17 00:00:00 2001
From: Joakim Hove <joakim.hove@gmail.com>
Date: Mon, 26 Apr 2021 14:14:58 +0200
Subject: [PATCH] Call endEpisode() for WellModel and Aquifer model

---
 ebos/eclproblem.hh | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/ebos/eclproblem.hh b/ebos/eclproblem.hh
index d8bccd94a..72ef82397 100644
--- a/ebos/eclproblem.hh
+++ b/ebos/eclproblem.hh
@@ -1255,8 +1255,11 @@ public:
         auto& simulator = this->simulator();
         auto& schedule = simulator.vanguard().schedule();
 
-        int episodeIdx = simulator.episodeIndex();
+        wellModel_.endEpisode();
+        if (enableAquifers_)
+            aquiferModel_.endEpisode();
 
+        int episodeIdx = simulator.episodeIndex();
         // check if we're finished ...
         if (episodeIdx + 1 >= static_cast<int>(schedule.size() - 1)) {
             simulator.setFinished(true);