From b9456b594609f95f52817ee437581fec17507c96 Mon Sep 17 00:00:00 2001 From: rvcas Date: Thu, 15 Aug 2024 12:45:33 -0400 Subject: [PATCH] fix: wrap_validator_condition needs to happen earlier --- crates/aiken-lang/src/gen_uplc.rs | 13 +++++-------- 1 file changed, 5 insertions(+), 8 deletions(-) diff --git a/crates/aiken-lang/src/gen_uplc.rs b/crates/aiken-lang/src/gen_uplc.rs index 233dbd5c..07115b2f 100644 --- a/crates/aiken-lang/src/gen_uplc.rs +++ b/crates/aiken-lang/src/gen_uplc.rs @@ -488,13 +488,10 @@ impl<'a> CodeGenerator<'a> { on_test_failure: OnTestFailure::FailImmediately, }; - let mut air_tree_fun = AirTree::anon_func( - vec!["__context__".to_string()], - self.build(&fun.body, module_name, &[]), - true, - ); + let air_tree_fun = + wrap_validator_condition(self.build(&fun.body, module_name, &[]), self.tracing); - air_tree_fun = wrap_validator_condition(air_tree_fun, self.tracing); + let air_tree_fun = AirTree::anon_func(vec!["__context__".to_string()], air_tree_fun, true); let validator_args_tree = AirTree::no_op(air_tree_fun); @@ -504,9 +501,9 @@ impl<'a> CodeGenerator<'a> { let full_vec = full_tree.to_vec(); - let mut term = self.uplc_code_gen(full_vec); + let term = self.uplc_code_gen(full_vec); - term = cast_validator_args(term, params); + let term = cast_validator_args(term, params); self.finalize(term) }