Fixed build to reflect updated API.

This should fix the build because the API has changed.