diff --git a/src/View/Results/PlotSedProfile.py b/src/View/Results/PlotSedProfile.py
index 895df719633913c52947433160ed2280e6bad16b..1db320f91c80c4bff5f786b677029be1fab020e3 100644
--- a/src/View/Results/PlotSedProfile.py
+++ b/src/View/Results/PlotSedProfile.py
@@ -51,13 +51,13 @@ class PlotSedProfile(PamhyrPlot):
         profiles_sl = list(
             map(
                 lambda sl: sl[0],
-                profile.get_ts_key(self._current_timestamp, "sl")
+                profile.get_ts_key(self._current_timestamp, "sl")[0]
             )
         )
         profiles_sl_0 = list(
             map(
                 lambda sl: sl[0],
-                profile.get_ts_key(0.0, "sl")
+                profile.get_ts_key(0.0, "sl")[0]
             )
         )
 
diff --git a/src/View/Results/PlotSedReach.py b/src/View/Results/PlotSedReach.py
index 3978930d2fb265ed60da62ce5b91bde6e2ccd2a9..b7f229c8c113a4425e96b4174e4a4957cad63b97 100644
--- a/src/View/Results/PlotSedReach.py
+++ b/src/View/Results/PlotSedReach.py
@@ -124,14 +124,14 @@ class PlotSedReach(PamhyrPlot):
         profiles_sl_0 = list(
             map(
                 # Get SL list for profile p at time 0 (initial data)
-                lambda p: p.get_ts_key(0.0, "sl"),
+                lambda p: p.get_ts_key(0.0, "sl")[0],
                 reach.profiles
             )
         )
         profiles_sl = list(
             map(
                 # Get SL list for profile p at current time
-                lambda p: p.get_ts_key(self._current_timestamp, "sl"),
+                lambda p: p.get_ts_key(self._current_timestamp, "sl")[0],
                 reach.profiles
             )
         )
@@ -163,7 +163,10 @@ class PlotSedReach(PamhyrPlot):
         z_sl = reduce(
             lambda acc, v: acc + [
                 list(
-                    map(lambda x, y: y - x, v, acc[-1])
+                    map(
+                        lambda x, y: y - x,
+                        v, acc[-1]
+                    )
                 )
             ],
             sl_0,