Fix the module name to match the class name