From 874f592cf4e6e36842ad04e15b82f81bf625dd05 Mon Sep 17 00:00:00 2001
From: Pierre-Antoine Rouby <contact@parouby.fr>
Date: Tue, 4 Nov 2025 14:18:00 +0100
Subject: [PATCH] Results: Fix add data settings.

---
 src/Model/Results/Results.py | 16 +++++++++++-----
 1 file changed, 11 insertions(+), 5 deletions(-)

diff --git a/src/Model/Results/Results.py b/src/Model/Results/Results.py
index c6e51e2e..aad79c04 100644
--- a/src/Model/Results/Results.py
+++ b/src/Model/Results/Results.py
@@ -116,15 +116,16 @@ class AdditionalData(SQLSubModel):
             y = struct.unpack(data_format, by)
 
             data = {
-                'type_x': tmp_dict[data_type[2]],
-                'type_y': tmp_dict[data_type[0]],
+                'type_x': type_x,
+                'type_y': type_y,
                 'legend': legend,
-                'unit': tmp_unit[data_type[0]],
+                'unit': unit,
                 'x': x, 'y': y
             }
 
-            new_results = cls(study=study)
-            new.append(new_results)
+            new_data = cls(study=study)
+            new_data._data = data
+            new.append(new_data)
 
         return new
 
@@ -310,6 +311,11 @@ class Results(SQLSubModel):
             data["timestamps"] = sorted(ts)
             new_results._river = River._db_load(execute, data)
 
+            new_results.set(
+                "additional_data",
+                AdditionalData._db_load(execute, data)
+            )
+
             yield (solver_type, new_results)
 
     def _db_save_clear(self, execute, solver_type, data=None):