diff --git a/pystencils/nbackend/ast/nodes.py b/pystencils/nbackend/ast/nodes.py
index 09774f90367d627a0c8e60b40215cf1d57368187..16a7bd29913f894487f92cf7a2c0aae8082ae9ad 100644
--- a/pystencils/nbackend/ast/nodes.py
+++ b/pystencils/nbackend/ast/nodes.py
@@ -38,8 +38,13 @@ class PsAstNode(ABC):
 
 
 class PsBlock(PsAstNode):
-    @PsAstNode.children.setter
-    def children(self, cs):
+
+    @property
+    def children(self) -> Generator[PsAstNode, None, None]:
+        yield from self._children  # need to override entire property to override the setter
+
+    @children.setter
+    def children(self, cs: Sequence[PsAstNode]):
         self._children = cs