From d52fcf0ec750a0b4ac2b2dd219f6940288e38f26 Mon Sep 17 00:00:00 2001
From: Alon Haramati <alonh@starkware.co>
Date: Thu, 20 Jun 2024 15:04:29 +0300
Subject: [PATCH] Air is AirTraceVerifier.

---
 crates/prover/src/core/air/mod.rs    | 2 +-
 crates/prover/src/core/prover/mod.rs | 4 ++--
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/crates/prover/src/core/air/mod.rs b/crates/prover/src/core/air/mod.rs
index de6e5e3d3..4b5f53c50 100644
--- a/crates/prover/src/core/air/mod.rs
+++ b/crates/prover/src/core/air/mod.rs
@@ -21,7 +21,7 @@ pub use air_ext::{AirExt, AirProverExt};
 /// For instance, all interaction elements are assumed to be present in it.
 /// Therefore, an AIR is generated only after the initial trace commitment phase.
 // TODO(spapini): consider renaming this struct.
-pub trait Air {
+pub trait Air: AirTraceVerifier {
     fn components(&self) -> Vec<&dyn Component>;
 }
 
diff --git a/crates/prover/src/core/prover/mod.rs b/crates/prover/src/core/prover/mod.rs
index 4ff3483da..4e18fb43c 100644
--- a/crates/prover/src/core/prover/mod.rs
+++ b/crates/prover/src/core/prover/mod.rs
@@ -2,7 +2,7 @@ use itertools::Itertools;
 use thiserror::Error;
 use tracing::{span, Level};
 
-use super::air::{AirProver, AirTraceVerifier, AirTraceWriter};
+use super::air::{AirProver, AirTraceWriter};
 use super::backend::Backend;
 use super::fields::secure_column::SECURE_EXTENSION_DEGREE;
 use super::fri::FriVerificationError;
@@ -195,7 +195,7 @@ pub fn prove<B: Backend + MerkleOps<MerkleHasher>>(
 
 pub fn verify(
     proof: StarkProof,
-    air: &(impl Air + AirTraceVerifier),
+    air: &impl Air,
     channel: &mut Channel,
 ) -> Result<(), VerificationError> {
     // Read trace commitment.