Fix a refactoring bug.