Fix bad refactoring.