Correct the module to take coverage for