Fix module loading function