test: simplify counts aggregation