Fix class exporting