diff --git a/python/ribasim/ribasim/model.py b/python/ribasim/ribasim/model.py
index 7a395ab5a..895cf7220 100644
--- a/python/ribasim/ribasim/model.py
+++ b/python/ribasim/ribasim/model.py
@@ -479,7 +479,7 @@ def _add_results(self, uds):
         # from node_id to the node_dim index
         node_lookup = pd.Series(
             index=uds["node_id"],
-            data=uds[edge_dim],
+            data=uds[node_dim],
             name="node_index",
         )
         # from edge_id to the edge_dim index