fix test with new lib