Tidy up the module loading