Minor fix for #9