Refactor correction to Corrector class