Fix module loading issue