Fixed the mods to work with latest NModule update