diff --git a/evm/src/witness/transition.rs b/evm/src/witness/transition.rs
index 55491bc0..e9b8c09b 100644
--- a/evm/src/witness/transition.rs
+++ b/evm/src/witness/transition.rs
@@ -182,8 +182,7 @@ fn fill_op_flag<F: Field>(op: Operation, row: &mut CpuColumnsView<F>) {
         Operation::Jump | Operation::Jumpi => &mut flags.jumps,
         Operation::Pc => &mut flags.pc,
         Operation::Jumpdest => &mut flags.jumpdest,
-        Operation::GetContext => &mut flags.context_op,
-        Operation::SetContext => &mut flags.context_op,
+        Operation::GetContext | Operation::SetContext => &mut flags.context_op,
         Operation::ExitKernel => &mut flags.exit_kernel,
         Operation::MloadGeneral => &mut flags.mload_general,
         Operation::MstoreGeneral => &mut flags.mstore_general,