Fix error created during refactoring.