Minor fix to correct return value.