Improve CountByReducerSpec