Fix typo. The dict is 'grains' not 'grain'