Minor fix to a bug