Minor fix to last change.