From 1974b0a172c1423e30d7f49b201d2e909bc03cbc Mon Sep 17 00:00:00 2001
From: hratoanina <hratoanina@polygon.technology>
Date: Mon, 29 Jul 2024 19:56:25 -0400
Subject: [PATCH] Fix Arithmetic trace length

---
 evm_arithmetization/src/witness/traces.rs | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/evm_arithmetization/src/witness/traces.rs b/evm_arithmetization/src/witness/traces.rs
index e70f36824..7dd973b2a 100644
--- a/evm_arithmetization/src/witness/traces.rs
+++ b/evm_arithmetization/src/witness/traces.rs
@@ -59,7 +59,12 @@ impl<T: Copy> Traces<T> {
                 .map(|op| match op {
                     Operation::TernaryOperation { .. } => 2,
                     Operation::BinaryOperation { operator, .. } => match operator {
-                        BinaryOperator::Div | BinaryOperator::Mod => 2,
+                        BinaryOperator::Div
+                        | BinaryOperator::Mod
+                        | BinaryOperator::AddFp254
+                        | BinaryOperator::SubFp254
+                        | BinaryOperator::MulFp254
+                        | BinaryOperator::Shr => 2,
                         _ => 1,
                     },
                     Operation::RangeCheckOperation { .. } => 1,