Support all_buckets field on test