Fixing module.exports issue