Fix silly bug during refactoring.