diff --git a/neural_lam/models/base_graph_model.py b/neural_lam/models/base_graph_model.py
index aee6837..91e6afb 100644
--- a/neural_lam/models/base_graph_model.py
+++ b/neural_lam/models/base_graph_model.py
@@ -43,7 +43,7 @@ def __init__(
             if name == "mesh_static_features":
                 max_coord = datastore.get_xy("state").max()
                 # Rescale by dividing by maximum coordinate in interior
-                attr_value = attr_value / max_coord
+                attr_value /= max_coord
 
             # Make BufferLists module members and register tensors as buffers
             if isinstance(attr_value, torch.Tensor):