Fix wrapper tests