Update tests to changed return value of `wrapWith()`