Fixing the module export