Minor fix for #18