Fix import of wrapper