fix for API change