Minor fix last commit