Fixed error introduced during refactoring.