Minor fix #2