Minor fix of previous commit