go/grains: Update to comply with new spec