Update to v0.3.1
Small fix to .wrap()