Update to new API for collection creation