Small fix for change left over from refactor.