Fixed test: dataset.groups is now a set, not list