Fix refactoring mistake.