diff --git a/src/Solver/Mage.py b/src/Solver/Mage.py
index 22168ea34178b7aee202b8572ea36dfa3b37284c..f3a358f143b46b1021eca33b67fc16891418a5c3 100644
--- a/src/Solver/Mage.py
+++ b/src/Solver/Mage.py
@@ -1009,10 +1009,8 @@ class Mage8(Mage):
                         # Set data for profile RI
                         reach.set(ri, timestamp, key, d)
                         if key == "Z":
-                            profile = study.river\
-                                           .current_reach()\
-                                           .reach.profile(ri)
-                            ptX, ptY = profile.get_water_limits(d)
+                            profile = reach.profile(ri)
+                            ptX, ptY = profile.geometry.get_water_limits(d)
                             reach.set(ri, timestamp, "ptX", ptX)
                             reach.set(ri, timestamp, "ptY", ptY)