Fix a place where I missed switching from grains to node groups