diff --git a/dev/aarch64_clean/src/arith_native_aarch64.h b/dev/aarch64_clean/src/arith_native_aarch64.h
index fc62ac2279..4633a14fb7 100644
--- a/dev/aarch64_clean/src/arith_native_aarch64.h
+++ b/dev/aarch64_clean/src/arith_native_aarch64.h
@@ -161,7 +161,7 @@ __contract__(
 #define mlk_rej_uniform_asm MLK_NAMESPACE(rej_uniform_asm)
 MLK_MUST_CHECK_RETURN_VALUE
 uint64_t mlk_rej_uniform_asm(int16_t r[256], const uint8_t *buf,
-                             unsigned buflen, const uint8_t table[2048])
+                             unsigned buflen, const uint8_t table[4096])
 /* This must be kept in sync with the HOL-Light specification
  * in proofs/hol_light/aarch64/proofs/mlkem_rej_uniform.ml. */
 __contract__(
diff --git a/dev/aarch64_clean/src/rej_uniform_asm.S b/dev/aarch64_clean/src/rej_uniform_asm.S
index 941ec842cd..be90e4b56b 100644
--- a/dev/aarch64_clean/src/rej_uniform_asm.S
+++ b/dev/aarch64_clean/src/rej_uniform_asm.S
@@ -6,7 +6,7 @@
 /*yaml
   Name: rej_uniform_asm
   Description: Run rejection sampling on uniform random bytes to generate uniform random integers mod q
-  Signature: uint64_t mlk_rej_uniform_asm(int16_t r[256], const uint8_t *buf, unsigned buflen, const uint8_t table[2048])
+  Signature: uint64_t mlk_rej_uniform_asm(int16_t r[256], const uint8_t *buf, unsigned buflen, const uint8_t table[4096])
   ABI:
     x0:
       type: buffer
@@ -27,9 +27,9 @@
       test_with: 504  # MLKEM_GEN_MATRIX_NBLOCKS * MLK_XOF_RATE
     x3:
       type: buffer
-      size_bytes: 2048
+      size_bytes: 4096
       permissions: read-only
-      c_parameter: const uint8_t table[2048]
+      c_parameter: const uint8_t table[4096]
       description: Lookup table
   Stack:
     bytes: 576
diff --git a/dev/aarch64_opt/src/arith_native_aarch64.h b/dev/aarch64_opt/src/arith_native_aarch64.h
index bce864a299..bf95bdaf9b 100644
--- a/dev/aarch64_opt/src/arith_native_aarch64.h
+++ b/dev/aarch64_opt/src/arith_native_aarch64.h
@@ -161,7 +161,7 @@ __contract__(
 #define mlk_rej_uniform_asm MLK_NAMESPACE(rej_uniform_asm)
 MLK_MUST_CHECK_RETURN_VALUE
 uint64_t mlk_rej_uniform_asm(int16_t r[256], const uint8_t *buf,
-                             unsigned buflen, const uint8_t table[2048])
+                             unsigned buflen, const uint8_t table[4096])
 /* This must be kept in sync with the HOL-Light specification
  * in proofs/hol_light/aarch64/proofs/mlkem_rej_uniform.ml. */
 __contract__(
diff --git a/dev/aarch64_opt/src/rej_uniform_asm.S b/dev/aarch64_opt/src/rej_uniform_asm.S
index f8cf2ed888..a31bb619eb 100644
--- a/dev/aarch64_opt/src/rej_uniform_asm.S
+++ b/dev/aarch64_opt/src/rej_uniform_asm.S
@@ -6,7 +6,7 @@
 /*yaml
   Name: rej_uniform_asm
   Description: Run rejection sampling on uniform random bytes to generate uniform random integers mod q
-  Signature: uint64_t mlk_rej_uniform_asm(int16_t r[256], const uint8_t *buf, unsigned buflen, const uint8_t table[2048])
+  Signature: uint64_t mlk_rej_uniform_asm(int16_t r[256], const uint8_t *buf, unsigned buflen, const uint8_t table[4096])
   ABI:
     x0:
       type: buffer
@@ -27,9 +27,9 @@
       test_with: 504  # MLKEM_GEN_MATRIX_NBLOCKS * MLK_XOF_RATE
     x3:
       type: buffer
-      size_bytes: 2048
+      size_bytes: 4096
       permissions: read-only
-      c_parameter: const uint8_t table[2048]
+      c_parameter: const uint8_t table[4096]
       description: Lookup table
   Stack:
     bytes: 576
diff --git a/mlkem/src/native/aarch64/src/arith_native_aarch64.h b/mlkem/src/native/aarch64/src/arith_native_aarch64.h
index 2941ecbd4b..a739cdf96f 100644
--- a/mlkem/src/native/aarch64/src/arith_native_aarch64.h
+++ b/mlkem/src/native/aarch64/src/arith_native_aarch64.h
@@ -161,7 +161,7 @@ __contract__(
 #define mlk_rej_uniform_asm MLK_NAMESPACE(rej_uniform_asm)
 MLK_MUST_CHECK_RETURN_VALUE
 uint64_t mlk_rej_uniform_asm(int16_t r[256], const uint8_t *buf,
-                             unsigned buflen, const uint8_t table[2048])
+                             unsigned buflen, const uint8_t table[4096])
 /* This must be kept in sync with the HOL-Light specification
  * in proofs/hol_light/aarch64/proofs/mlkem_rej_uniform.ml. */
 __contract__(
diff --git a/mlkem/src/native/aarch64/src/rej_uniform_asm.S b/mlkem/src/native/aarch64/src/rej_uniform_asm.S
index b03139b46a..fd9482813d 100644
--- a/mlkem/src/native/aarch64/src/rej_uniform_asm.S
+++ b/mlkem/src/native/aarch64/src/rej_uniform_asm.S
@@ -6,7 +6,7 @@
 /*yaml
   Name: rej_uniform_asm
   Description: Run rejection sampling on uniform random bytes to generate uniform random integers mod q
-  Signature: uint64_t mlk_rej_uniform_asm(int16_t r[256], const uint8_t *buf, unsigned buflen, const uint8_t table[2048])
+  Signature: uint64_t mlk_rej_uniform_asm(int16_t r[256], const uint8_t *buf, unsigned buflen, const uint8_t table[4096])
   ABI:
     x0:
       type: buffer
@@ -27,9 +27,9 @@
       test_with: 504  # MLKEM_GEN_MATRIX_NBLOCKS * MLK_XOF_RATE
     x3:
       type: buffer
-      size_bytes: 2048
+      size_bytes: 4096
       permissions: read-only
-      c_parameter: const uint8_t table[2048]
+      c_parameter: const uint8_t table[4096]
       description: Lookup table
   Stack:
     bytes: 576
diff --git a/proofs/hol_light/aarch64/mlkem/mlkem_rej_uniform.S b/proofs/hol_light/aarch64/mlkem/mlkem_rej_uniform.S
index 7f3dfa55d3..ccfff15465 100644
--- a/proofs/hol_light/aarch64/mlkem/mlkem_rej_uniform.S
+++ b/proofs/hol_light/aarch64/mlkem/mlkem_rej_uniform.S
@@ -6,7 +6,7 @@
 /*yaml
   Name: rej_uniform_asm
   Description: Run rejection sampling on uniform random bytes to generate uniform random integers mod q
-  Signature: uint64_t mlk_rej_uniform_asm(int16_t r[256], const uint8_t *buf, unsigned buflen, const uint8_t table[2048])
+  Signature: uint64_t mlk_rej_uniform_asm(int16_t r[256], const uint8_t *buf, unsigned buflen, const uint8_t table[4096])
   ABI:
     x0:
       type: buffer
@@ -27,9 +27,9 @@
       test_with: 504  # MLKEM_GEN_MATRIX_NBLOCKS * MLK_XOF_RATE
     x3:
       type: buffer
-      size_bytes: 2048
+      size_bytes: 4096
       permissions: read-only
-      c_parameter: const uint8_t table[2048]
+      c_parameter: const uint8_t table[4096]
       description: Lookup table
   Stack:
     bytes: 576