Fix a bad refactor.