diff --git a/pystencils/interpolation_astnodes.py b/pystencils/interpolation_astnodes.py index 79e534b6ec7dde7c974173e0b184110e49fd2c52..27b11948165846f6e218bece94d89b07319463da 100644 --- a/pystencils/interpolation_astnodes.py +++ b/pystencils/interpolation_astnodes.py @@ -172,7 +172,7 @@ class InterpolatorAccess(TypedSymbol): offsets.update([self]) for o in self.offsets: if hasattr(o, 'atoms'): - offsets.update(set(o.atoms(types))) + offsets.update(set(o.atoms(*types))) return offsets else: return set()