fix compatibility with new version