diff --git a/src/anemoi/graphs/edges/attributes.py b/src/anemoi/graphs/edges/attributes.py
index 3d8122e..3c1a21e 100644
--- a/src/anemoi/graphs/edges/attributes.py
+++ b/src/anemoi/graphs/edges/attributes.py
@@ -202,7 +202,7 @@ def get_raw_values(self, graph: HeteroData, source_name: str, target_name: str)
 
         edge_index = graph[(source_name, "to", target_name)].edge_index
         assert hasattr(graph[node_name], self.node_attr_name)
-        val = getattr(graph[node_name], self.node_attr_name).numpy()[edge_index[self.idx]]
+        return graph[node_name][self.node_attr_name].numpy()[edge_index[self.idx]]
         return val