Add a few tests for wrap