Update project for API change