diff --git a/crates/brainfuck_prover/src/components/memory/component.rs b/crates/brainfuck_prover/src/components/memory/component.rs
index 1fd52ca..8df7cbc 100644
--- a/crates/brainfuck_prover/src/components/memory/component.rs
+++ b/crates/brainfuck_prover/src/components/memory/component.rs
@@ -1,9 +1,4 @@
-use super::table::MemoryColumn;
-use stwo_prover::core::{
-    channel::Channel,
-    fields::{qm31::SecureField, secure_column::SECURE_EXTENSION_DEGREE},
-    pcs::TreeVec,
-};
+use stwo_prover::core::{channel::Channel, fields::qm31::SecureField};
 
 /// The claim of the interaction phase 2 (with the logUp protocol).
 ///
diff --git a/crates/brainfuck_prover/src/components/memory/table.rs b/crates/brainfuck_prover/src/components/memory/table.rs
index 3912ddf..90ee38d 100644
--- a/crates/brainfuck_prover/src/components/memory/table.rs
+++ b/crates/brainfuck_prover/src/components/memory/table.rs
@@ -267,11 +267,16 @@ impl MemoryColumn {
             Self::D => 3,
         }
     }
+
+    /// Returns the total number of columns in the Memory table
+    pub const fn column_count() -> usize {
+        4
+    }
 }
 
 impl TraceColumn for MemoryColumn {
     fn count() -> usize {
-        4
+        Self::column_count()
     }
 }
 
@@ -284,7 +289,7 @@ impl TraceColumn for MemoryColumn {
 /// There are 3 lookup elements in the Memory component, as only the 'real' registers
 /// are used: `clk`, `mp` and `mv`. `d` is used to eventually nullify the numerator.
 #[derive(Clone, Debug, Eq, PartialEq)]
-pub struct MemoryElements(LookupElements<{ MemoryColumn::count() - 1 }>);
+pub struct MemoryElements(LookupElements<{ MemoryColumn::column_count() - 1 }>);
 
 impl MemoryElements {
     /// Provides dummy lookup elements.