From 97f0f40e7d8def5d1bfd379e8b409f3b4f601b04 Mon Sep 17 00:00:00 2001
From: Ivan Mikushin <imikushin@users.noreply.github.com>
Date: Mon, 21 Oct 2024 14:38:20 -0700
Subject: [PATCH] Fix test broken after branch update

---
 jolt-core/src/jolt/vm/rv32i_vm.rs | 15 +++++++++------
 1 file changed, 9 insertions(+), 6 deletions(-)

diff --git a/jolt-core/src/jolt/vm/rv32i_vm.rs b/jolt-core/src/jolt/vm/rv32i_vm.rs
index 6c14ddf03..ba09b6787 100644
--- a/jolt-core/src/jolt/vm/rv32i_vm.rs
+++ b/jolt-core/src/jolt/vm/rv32i_vm.rs
@@ -497,12 +497,15 @@ mod tests {
 
         let preprocessing =
             RV32IJoltVM::preprocess(bytecode.clone(), memory_init, 1 << 20, 1 << 20, 1 << 20);
-        let (proof, commitments, debug_info) =
-            <RV32IJoltVM as Jolt<Fr, HyperKZG<Bn254>, C, M>>::prove(
-                io_device,
-                trace,
-                preprocessing.clone(),
-            );
+        let (proof, commitments, debug_info) = <RV32IJoltVM as Jolt<
+            Fr,
+            HyperKZG<Bn254, KeccakTranscript>,
+            C,
+            M,
+            KeccakTranscript,
+        >>::prove(
+            io_device, trace, preprocessing.clone()
+        );
         let verification_result =
             RV32IJoltVM::verify(preprocessing, proof, commitments, debug_info);
         assert!(