diff --git a/vp9/common/vp9_onyxc_int.h b/vp9/common/vp9_onyxc_int.h
index 8f957345d5a652d8fd0af975d6249c1d0fe12c41..84cb9eff3fa531cf03eee3644545c5be0b2aa296 100644
--- a/vp9/common/vp9_onyxc_int.h
+++ b/vp9/common/vp9_onyxc_int.h
@@ -361,6 +361,10 @@ static int get_mb_col(const MACROBLOCKD *xd) {
 }
 
 static int get_token_alloc(int mb_rows, int mb_cols) {
+#if CONFIG_CODE_ZEROGROUP
+  return mb_rows * mb_cols * (24 * 16 * 2);
+#else
   return mb_rows * mb_cols * (24 * 16 + 4);
+#endif
 }
 #endif  // VP9_COMMON_VP9_ONYXC_INT_H_