diff --git a/src/Model/River.py b/src/Model/River.py
index acda6e22..58d93cd5 100644
--- a/src/Model/River.py
+++ b/src/Model/River.py
@@ -364,7 +364,7 @@ class RiverReach(Edge):
             )
             new.enable(enable=enabled)
             if deleted:
-                nd.set_as_deleted()
+                new.set_as_deleted()
 
             data["reach"] = new
             new._reach = Reach._db_load(execute, data)
@@ -549,7 +549,7 @@ class River(Graph):
         new._nodes = RiverNode._db_load(
             execute, data
         )
-        data["nodes"] = new.nodes()
+        data["nodes"] = new._nodes
         data['loaded_pid'] = set()
 
         new._edges = RiverReach._db_load(
diff --git a/src/Model/Tools/PamhyrDB.py b/src/Model/Tools/PamhyrDB.py
index d1706f2c..9ca7257c 100644
--- a/src/Model/Tools/PamhyrDB.py
+++ b/src/Model/Tools/PamhyrDB.py
@@ -203,7 +203,7 @@ class SQLSubModel(PamhyrID):
         if self._status is None:
             return
 
-        # self._owner_scenario = self._status.scenario_id
+        self._owner_scenario = self._status.scenario_id
         self._status.modified()
 
     def set_owner_scenario(self):