fix test with new wrapper