diff --git a/pytato/transform.py b/pytato/transform.py
index 2e252df3970364fb313e6258c19144ac8b979f88..b4192654818759ecc54da6dba780dc30202a3924 100644
--- a/pytato/transform.py
+++ b/pytato/transform.py
@@ -500,6 +500,8 @@ class WalkMapper(Mapper):
             if isinstance(dim, Array):
                 self.rec(dim)
 
+        self.post_visit(expr)
+
     def map_loopy_call(self, expr: LoopyCall) -> None:
         if not self.visit(expr):
             return