Fix module loading

@quantum5