fix(build): Set up correct modules in build