Fix bug that prevents module from exporting correctly