Fixing module code