Changed tests for new version