Use __grains__ properly instead of looping through items()