Fix modifyAllMatching visitor helper (#5020)

Signed-off-by: Vladimír Štill <vladimir.still@intel.com>
p4lang · Nov 19, 2024 · 9bef7ee · 9bef7ee
1 parent 929c928
commit 9bef7ee
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ir/visitor.h b/ir/visitor.h
@@ -820,7 +820,7 @@ const RootType *modifyAllMatching(const RootType *root, Func &&function) {
         Func function;
         void postorder(NodeType *node) override { function(node); }
     };
-    return root->apply(NodeVisitor(std::forward<Func>(function)));
+    return root->apply(NodeVisitor(std::forward<Func>(function)))->template checkedTo<RootType>();
 }
 
 /**