Minor fix from last commit