Minor fix of last commit