Minor fix 2