Add .collapse() to stop group-by chaining