Set corpus size in generator