Fix wrapper return value