Fix tests for wrap