diff --git a/tests/benchmark/compute/instruction/test_keccak.py b/tests/benchmark/compute/instruction/test_keccak.py
index f117a173232..7211d04794f 100644
--- a/tests/benchmark/compute/instruction/test_keccak.py
+++ b/tests/benchmark/compute/instruction/test_keccak.py
@@ -103,6 +103,6 @@ def test_keccak_diff_mem_msg_sizes(
         target_opcode=Op.SHA3,
         code_generator=JumpLoopGenerator(
             setup=setup,
-            attack_block=Op.POP(Op.SHA3(0, msg_size)),
+            attack_block=Op.MSTORE(Op.PUSH0, Op.SHA3(Op.PUSH0, msg_size)),
         ),
     )