Fix custom group test