Updated source to reflect recent GRAIL API change.

Change-Id: I62530199e365268a47b459028c443500f4b36b63