Fixes the module loading