Minor fix for grouping by test