Fix for minor API change.