Fix the module add feature