Fix module name mismatch