diff --git a/main.py b/main.py
index 34f2bb3..d0a33d7 100644
--- a/main.py
+++ b/main.py
@@ -31,7 +31,7 @@ def main():
     if len(msg) != 40:
         raise ValueError("Message must be exactly 40 characters.")
     num_blocks = len(msg)
-    C = make_codebook(r=5, num_blocks=num_blocks, margin=0.95)
+    C = make_codebook(r=6, num_blocks=num_blocks)
     x = encode_message(msg, C)
     print(f"→ Total samples = {x.size}, total energy = {np.sum(x*x):.1f}")