diff --git a/src/pystencilssfg/lang/expressions.py b/src/pystencilssfg/lang/expressions.py index b3ed18e276c2485702247a651d3d9a7c593d594c..72287eaad73afc08770d451c0847c362ef561519 100644 --- a/src/pystencilssfg/lang/expressions.py +++ b/src/pystencilssfg/lang/expressions.py @@ -7,7 +7,7 @@ import sympy as sp from pystencils import TypedSymbol from pystencils.codegen import Parameter -from pystencils.types import PsType, PsScalarType, UserTypeSpec, create_type +from pystencils.types import PsType, PsIntegerType, UserTypeSpec, create_type from ..exceptions import SfgException from .headers import HeaderFile @@ -474,7 +474,7 @@ def includes(obj: ExprLike | PsType) -> set[HeaderFile]: case PsType(): headers = set(HeaderFile.parse(h) for h in obj.required_headers) - if isinstance(obj, PsScalarType): + if isinstance(obj, PsIntegerType): headers.add(HeaderFile.parse("<cstdint>")) return headers