Fix bad refactor.