Minor fix to account for earlier refactoring.