Fix module mapping