Minor fix to validation