Fix NestBy extension signature