diff --git a/evm/src/cpu/kernel/tests/sha2.rs b/evm/src/cpu/kernel/tests/sha2.rs
index d0c21090..cb6d580a 100644
--- a/evm/src/cpu/kernel/tests/sha2.rs
+++ b/evm/src/cpu/kernel/tests/sha2.rs
@@ -32,7 +32,7 @@ fn test_sha2() -> Result<()> {
     initial_stack.reverse();
 
     // Run the sha2 kernel code.
-    let result = run(&kernel.code, sha2, initial_stack, &kernel.prover_inputs)?;
+    let result = run_with_kernel(&kernel, sha2, initial_stack)?;
     let actual = result.stack()[0];
 
     // Check that the result is correct.