diff --git a/kimchi/src/circuits/polynomials/keccak/witness.rs b/kimchi/src/circuits/polynomials/keccak/witness.rs
index 5161c4a708..a22713039f 100644
--- a/kimchi/src/circuits/polynomials/keccak/witness.rs
+++ b/kimchi/src/circuits/polynomials/keccak/witness.rs
@@ -19,9 +19,9 @@ use super::{
     RATE,
 };
 
-type Layout<const W: usize, F> = Vec<Box<dyn WitnessCell<W, F, Vec<F>>>>;
+type Layout<F, const COLUMNS: usize> = Vec<Box<dyn WitnessCell<F, Vec<F>, COLUMNS>>>;
 
-fn layout_round<F: PrimeField>() -> [Layout<KECCAK_COLS, F>; 1] {
+fn layout_round<F: PrimeField>() -> [Layout<F, KECCAK_COLS>; 1] {
     [vec![
         IndexCell::create("state_a", 0, 100),
         IndexCell::create("state_c", 100, 120),
@@ -48,7 +48,7 @@ fn layout_round<F: PrimeField>() -> [Layout<KECCAK_COLS, F>; 1] {
     ]]
 }
 
-fn layout_sponge<F: PrimeField>() -> [Layout<KECCAK_COLS, F>; 1] {
+fn layout_sponge<F: PrimeField>() -> [Layout<F, KECCAK_COLS>; 1] {
     [vec![
         IndexCell::create("old_state", 0, 100),
         IndexCell::create("new_state", 100, 200),