Fix a refactoring error