Minor fix up