Tidy up module loader code