diff --git a/configs/example/apu_se.py b/configs/example/apu_se.py
index 29bb452c69..ffd2578e3d 100644
--- a/configs/example/apu_se.py
+++ b/configs/example/apu_se.py
@@ -249,6 +249,11 @@ parser.add_argument(
     action="store_true",
     help="Count Page Accesses and output in per-CU output files",
 )
+parser.add_argument(
+    "--max-cu-tokens",
+    type=int,
+    help="Number of coalescer tokens per CU",
+)
 parser.add_argument(
     "--TLB-prefetch", type=int, help="prefetch depth for" "TLBs"
 )
@@ -433,6 +438,7 @@ for i in range(n_cu):
             functionalTLB=args.FunctionalTLB,
             localMemBarrier=args.LocalMemBarrier,
             countPages=args.countPages,
+            max_cu_tokens=args.max_cu_tokens,
             localDataStore=LdsState(
                 banks=args.numLdsBanks,
                 bankConflictPenalty=args.ldsBankConflictPenalty,