Update the way the module is exported