fix test after wrapping change