diff --git a/pytential/symbolic/execution.py b/pytential/symbolic/execution.py index a4ec057a8..8a5d2f4b9 100644 --- a/pytential/symbolic/execution.py +++ b/pytential/symbolic/execution.py @@ -252,7 +252,7 @@ def map_ones(self, expr: pp.Ones): self.array_context.thaw(discr.nodes()[0])) def map_node_coordinate_component(self, expr: pp.NodeCoordinateComponent): - discr = self.places.get_discretization( + discr = self.places.get_target_or_discretization( expr.dofdesc.geometry, expr.dofdesc.discr_stage) x = discr.nodes()[expr.ambient_axis]