Updated to reflect API change