update tests to work with new version