Fixed build to reflect updated API.