Fix a bad refactor