diff --git a/src/Solver/Mage.py b/src/Solver/Mage.py
index f1711b70..defbdedd 100644
--- a/src/Solver/Mage.py
+++ b/src/Solver/Mage.py
@@ -558,12 +558,12 @@ class Mage(CommandLineSolver):
                 if not hs.enabled:
                     continue
 
-                if hs.input_rk is None:
+                if hs.input_section is None:
                     continue
 
                 f.write(
                     '* ouvrage au pk ' +
-                    f"{float(hs.input_rk):>12.1f}" + ' ' +
+                    f"{float(hs.input_section.rk):>12.1f}" + ' ' +
                     hs.name + '\n'
                 )
 
@@ -590,7 +590,7 @@ class Mage(CommandLineSolver):
 
                 f.write(
                     f"{sin_dict[bhs._type]} " +
-                    f"{reach_id} {float(hs.input_rk):>12.3f} " +
+                    f"{reach_id} {float(hs.input_section.rk):>12.3f} " +
                     f"{param_str} {name}\n"
                 )
 
diff --git a/src/View/HydraulicStructures/BasicHydraulicStructures/Window.py b/src/View/HydraulicStructures/BasicHydraulicStructures/Window.py
index 1e267706..7030360c 100644
--- a/src/View/HydraulicStructures/BasicHydraulicStructures/Window.py
+++ b/src/View/HydraulicStructures/BasicHydraulicStructures/Window.py
@@ -160,9 +160,9 @@ class BasicHydraulicStructuresWindow(PamhyrWindow):
         self.plot_layout.addWidget(self.canvas)
 
         reach = self._hs.input_reach
-        profile_rk = self._hs.input_rk
-        if profile_rk is not None:
-            profiles = reach.reach.get_profiles_from_rk(float(profile_rk))
+        profile = self._hs.input_section
+        if profile is not None:
+            profiles = reach.reach.get_profiles_from_rk(float(profile.rk))
         else:
             profiles = None