add correct module loading