Fixes some code comments to reflect API changes.