1 files changed, 17 insertions, 0 deletions
diff --git a/Source/Core/LambdaHelper.ssc b/Source/Core/LambdaHelper.ssc
index 218379b0..716ea7ae 100644
--- a/Source/Core/LambdaHelper.ssc
+++ b/Source/Core/LambdaHelper.ssc
@@ -41,6 +41,23 @@ namespace Microsoft.Boogie {
       internal List<Function!>! lambdaFunctions = new List<Function!>();
       static int lambdaid = 0;
 
+      public override Program! VisitProgram(Program! prog)
+      {
+        foreach (Declaration! decl in prog.TopLevelDeclarations) {
+          if (decl is Axiom) {
+            this.Visit(decl);
+          }
+        }
+
+        return prog;
+      }
+      
+      public override Procedure! VisitProcedure(Procedure! node)
+      {
+        // do not visit requires/ensures when calling this on Implementation
+        return node;
+      }  
+
       public override Absy! Visit(Absy! node)
       {
         node = base.Visit(node);