fix tests for .wrap