output hash fix

crates/air/src/layout/dex/mod.rs

@@ -395,8 +395,10 @@ impl LayoutTrait for Layout {
         let program_hash = pedersen_hash(&hash, &Felt::from(program.len()));
 
         let output_len: usize = (output_stop - output_start).to_bigint().try_into().unwrap();
-        let output = &memory[memory.len() - output_len..];
-        let output_hash = output.iter().fold(Felt::ZERO, |acc, e| pedersen_hash(&acc, e));
+        let output = &memory[memory.len() - output_len * 2..];
+        let hash =
+            output.iter().skip(1).step_by(2).fold(Felt::ZERO, |acc, e| pedersen_hash(&acc, e));
+        let output_hash = pedersen_hash(&hash, &Felt::from(output_len));
 
         Ok((program_hash, output_hash))
     }

crates/air/src/layout/recursive/mod.rs

@@ -400,8 +400,10 @@ impl LayoutTrait for Layout {
         let program_hash = pedersen_hash(&hash, &Felt::from(program.len()));
 
         let output_len: usize = (output_stop - output_start).to_bigint().try_into().unwrap();
-        let output = &memory[memory.len() - output_len..];
-        let output_hash = output.iter().fold(Felt::ZERO, |acc, e| pedersen_hash(&acc, e));
+        let output = &memory[memory.len() - output_len * 2..];
+        let hash =
+            output.iter().skip(1).step_by(2).fold(Felt::ZERO, |acc, e| pedersen_hash(&acc, e));
+        let output_hash = pedersen_hash(&hash, &Felt::from(output_len));
 
         Ok((program_hash, output_hash))
     }

crates/air/src/layout/recursive_with_poseidon/mod.rs

@@ -442,8 +442,10 @@ impl LayoutTrait for Layout {
         let program_hash = pedersen_hash(&hash, &Felt::from(program.len()));
 
         let output_len: usize = (output_stop - output_start).to_bigint().try_into().unwrap();
-        let output = &memory[memory.len() - output_len..];
-        let output_hash = output.iter().fold(Felt::ZERO, |acc, e| pedersen_hash(&acc, e));
+        let output = &memory[memory.len() - output_len * 2..];
+        let hash =
+            output.iter().skip(1).step_by(2).fold(Felt::ZERO, |acc, e| pedersen_hash(&acc, e));
+        let output_hash = pedersen_hash(&hash, &Felt::from(output_len));
 
         Ok((program_hash, output_hash))
     }

crates/air/src/layout/small/mod.rs

@@ -395,8 +395,10 @@ impl LayoutTrait for Layout {
         let program_hash = pedersen_hash(&hash, &Felt::from(program.len()));
 
         let output_len: usize = (output_stop - output_start).to_bigint().try_into().unwrap();
-        let output = &memory[memory.len() - output_len..];
-        let output_hash = output.iter().fold(Felt::ZERO, |acc, e| pedersen_hash(&acc, e));
+        let output = &memory[memory.len() - output_len * 2..];
+        let hash =
+            output.iter().skip(1).step_by(2).fold(Felt::ZERO, |acc, e| pedersen_hash(&acc, e));
+        let output_hash = pedersen_hash(&hash, &Felt::from(output_len));
 
         Ok((program_hash, output_hash))
     }

crates/air/src/layout/starknet/mod.rs

@@ -484,8 +484,10 @@ impl LayoutTrait for Layout {
         let program_hash = pedersen_hash(&hash, &Felt::from(program.len()));
 
         let output_len: usize = (output_stop - output_start).to_bigint().try_into().unwrap();
-        let output = &memory[memory.len() - output_len..];
-        let output_hash = output.iter().fold(Felt::ZERO, |acc, e| pedersen_hash(&acc, e));
+        let output = &memory[memory.len() - output_len * 2..];
+        let hash =
+            output.iter().skip(1).step_by(2).fold(Felt::ZERO, |acc, e| pedersen_hash(&acc, e));
+        let output_hash = pedersen_hash(&hash, &Felt::from(output_len));
 
         Ok((program_hash, output_hash))
     }

crates/air/src/layout/starknet_with_keccak/mod.rs

@@ -1,8 +1,6 @@
 pub mod autogenerated;
 pub mod global_values;
 
-use std::println;
-
 use crate::{
     diluted::get_diluted_product,
     layout::stark_curve,