Minor fix to warning